Run tests on T4

Author: VikParuchuri

.github/workflows/benchmarks.yml

@@ -7,10 +7,7 @@ env:
 
 jobs:
   build:
-    runs-on: ${{ matrix.os }}
-    strategy:
-      matrix:
-        os: [ubuntu-latest, windows-latest]
+    runs-on: t4_gpu
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python 3.11

.github/workflows/ci.yml

@@ -2,12 +2,12 @@ name: Unit tests
 
 on: [push]
 
-env:
-  TORCH_DEVICE: "cpu"
-
 jobs:
   build:
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [t4_gpu, ubuntu-latest, windows-latest]
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python 3.11

.github/workflows/scripts.yml

@@ -2,12 +2,9 @@ name: Test CLI scripts
 
 on: [push]
 
-env:
-  TORCH_DEVICE: "cpu"
-
 jobs:
   build:
-    runs-on: ubuntu-latest
+    runs-on: t4_gpu
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python 3.11