removed cuda device

GATEOverflow · Oct 8, 2024 · 1711455 · 1711455
1 parent 7b8cacf
commit 1711455
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test-mlperf-inference-mixtral.yml b/.github/workflows/test-mlperf-inference-mixtral.yml
@@ -16,7 +16,7 @@ jobs:
       matrix:
         python-version: [ "3.12" ]
         backend: [ "pytorch" ]
-        device: [ "cpu", "cuda" ]
+        device: [ "cpu" ]
 
     steps:
     - name: Test MLPerf Inference MIXTRAL-8X7B reference implementation