diff --git a/.github/workflows/userbenchmark-a100-release.yml b/.github/workflows/userbenchmark-a100-release.yml index 6523af9691..803f8ed549 100644 --- a/.github/workflows/userbenchmark-a100-release.yml +++ b/.github/workflows/userbenchmark-a100-release.yml @@ -3,16 +3,16 @@ on: pull_request: paths: - userbenchmark/release-test/* + - .github/workflows/userbenchmark-a100-release.yml jobs: run-userbenchmark: - runs-on: [a100-runner] + runs-on: linux.aws.a100 timeout-minutes: 1440 # 24 hours environment: docker-s3-upload env: BASE_CONDA_ENV: "torchbench" CONDA_ENV: "userbenchmark-a100" - PLATFORM_NAME: "gcp_a100" SETUP_SCRIPT: "/workspace/setup_instance.sh" steps: - name: Checkout TorchBench @@ -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}"