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