From 445174f398caca6272946e1b26fd943a2f996666 Mon Sep 17 00:00:00 2001
From: Samuel Burnham <45365069+samuelburnham@users.noreply.github.com>
Date: Thu, 26 Oct 2023 18:28:29 -0400
Subject: [PATCH] Temporary bench action test (#4)

* Temporary bench action test

* Test T4 GPU
---
 .github/workflows/bench_pr_comment.yml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/bench_pr_comment.yml b/.github/workflows/bench_pr_comment.yml
index 0c59daa51b..87285bf782 100644
--- 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