Temporary bench action test (#4)

* Temporary bench action test * Test T4 GPU
samuelburnham · Oct 26, 2023 · 445174f · 445174f
1 parent be2e2a1
commit 445174f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/bench_pr_comment.yml b/.github/workflows/bench_pr_comment.yml
@@ -39,9 +39,10 @@ jobs:
           # Needed. The name of the branch to compare with
           branchName: ${{ github.ref_name }}
 
+  # TODO: Test timings with different architectures
   gpu-benchmark:
     name: run fibonacci benchmark on GPU
-    runs-on: [self-hosted, gpu-bench-aws]
+    runs-on: [self-hosted, gpu-bench-t4]
     env:
       NVIDIA_VISIBLE_DEVICES: all
       NVIDIA_DRIVER_CAPABILITITES: compute,utility