diff --git a/.github/workflows/userbenchmark-a100-bisection.yml b/.github/workflows/userbenchmark-a100-bisection.yml index 8f25e3919..e133f194c 100644 --- a/.github/workflows/userbenchmark-a100-bisection.yml +++ b/.github/workflows/userbenchmark-a100-bisection.yml @@ -28,7 +28,7 @@ jobs: AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }} HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} if: ${{ github.repository_owner == 'pytorch' }} - runs-on: [a100-runner] + runs-on: linux.aws.a100 timeout-minutes: 2880 # 48 hours steps: - name: Checkout @@ -53,11 +53,6 @@ jobs: repository: pytorch/audio path: srcs/audio fetch-depth: 0 - - name: Tune Nvidia GPU - run: | - sudo nvidia-smi -pm 1 - sudo nvidia-smi -ac 1215,1410 - nvidia-smi - name: Install devel packages run: | sudo apt-get update -y diff --git a/.github/workflows/userbenchmark-a100-release.yml b/.github/workflows/userbenchmark-a100-release.yml index 6523af969..90333dddc 100644 --- a/.github/workflows/userbenchmark-a100-release.yml +++ b/.github/workflows/userbenchmark-a100-release.yml @@ -6,7 +6,7 @@ on: jobs: run-userbenchmark: - runs-on: [a100-runner] + runs-on: linux.aws.a100 timeout-minutes: 1440 # 24 hours environment: docker-s3-upload env: @@ -20,11 +20,6 @@ jobs: with: path: benchmark ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} - - name: Tune Nvidia GPU - run: | - sudo nvidia-smi -pm 1 - sudo nvidia-smi -ac 1215,1410 - nvidia-smi - name: Clone and setup conda env run: | CONDA_ENV=${BASE_CONDA_ENV} . "${SETUP_SCRIPT}" diff --git a/.github/workflows/userbenchmark-a100.yml b/.github/workflows/userbenchmark-a100.yml index 67d4003bd..70f8235ef 100644 --- a/.github/workflows/userbenchmark-a100.yml +++ b/.github/workflows/userbenchmark-a100.yml @@ -11,7 +11,7 @@ on: jobs: run-userbenchmark: - runs-on: [a100-runner] + runs-on: linux.aws.a100 timeout-minutes: 1440 # 24 hours environment: docker-s3-upload env: @@ -27,11 +27,6 @@ jobs: uses: actions/checkout@v3 with: path: benchmark - - name: Tune Nvidia GPU - run: | - sudo nvidia-smi -pm 1 - sudo nvidia-smi -ac 1215,1410 - nvidia-smi - name: Clone and setup conda env run: | CONDA_ENV=${BASE_CONDA_ENV} . "${SETUP_SCRIPT}" diff --git a/.github/workflows/v3-bisection.yml b/.github/workflows/v3-bisection.yml index 3d9541791..e68cbee80 100644 --- a/.github/workflows/v3-bisection.yml +++ b/.github/workflows/v3-bisection.yml @@ -20,7 +20,7 @@ jobs: AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }} AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }} if: ${{ github.repository_owner == 'pytorch' }} - runs-on: [a100-runner] + runs-on: linux.aws.a100 timeout-minutes: 2880 # 48 hours steps: - name: Checkout @@ -51,11 +51,6 @@ jobs: repository: pytorch/audio path: srcs/audio fetch-depth: 0 - - name: Tune Nvidia GPU - run: | - sudo nvidia-smi -pm 1 - sudo nvidia-smi -ac 1215,1410 - nvidia-smi - name: Install Deps run: | sudo apt-get -y update && sudo apt -y update