And turn nvidia-smi back on

triton-lang · Oct 16, 2023 · 4ac8f7c · 4ac8f7c
1 parent b2835e2
commit 4ac8f7c
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/.github/workflows/torchbench.yml b/.github/workflows/torchbench.yml
@@ -43,17 +43,17 @@ jobs:
           df -h
           du --max-depth=1 -h
 
-      # - name: Tune Nvidia GPU
-      #   run: |
-      #     sudo nvidia-smi -pm 1
-      #     sudo nvidia-smi --lock-gpu-clocks=1280,1280
-      #     nvidia-smi
+      - name: Tune Nvidia GPU
+        run: |
+          sudo nvidia-smi -pm 1
+          sudo nvidia-smi --lock-gpu-clocks=1280,1280
+          nvidia-smi
 
       - name: Run TorchBench
         run: |
           cd torchbench
           python3 test.py -v TestBenchmark.test_detectron2_fasterrcnn_r_101_c4_train_cuda
 
-      # - name: Reset GPU clocks
-      #   run: |
-      #     sudo nvidia-smi -i 0 -rgc
+      - name: Reset GPU clocks
+        run: |
+          sudo nvidia-smi -i 0 -rgc