Fix a100 bisect

pytorch · Nov 24, 2023 · fcf9b7d · fcf9b7d
1 parent 8e5e3a0
commit fcf9b7d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/userbenchmark-a100-bisection.yml b/.github/workflows/userbenchmark-a100-bisection.yml
@@ -72,6 +72,8 @@ jobs:
           python utils/cuda_utils.py --install-torchbench-deps
           mkdir -p "${BISECT_WORKDIR}"
           python utils/cuda_utils.py --install-torch-nightly
+          # The userbenchmark requires psutil
+          pip install psutil
           python run_benchmark.py ${{ github.event.inputs.userbenchmark }} ${{ github.event.inputs.userbenchmark_args }} --dryrun \
                  --output "${BISECT_WORKDIR}/metric-control.json"
           python run_benchmark.py ${{ github.event.inputs.userbenchmark }} ${{ github.event.inputs.userbenchmark_args }} --dryrun \