updations for amd - docs

GATEOverflow · Oct 21, 2024 · 7eee072 · 7eee072
1 parent 148650d
commit 7eee072
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -44,8 +44,9 @@ def mlperf_inference_implementation_readme(spaces, model, implementation, *, imp
             frameworks = [ "TensorRT" ]
 
         elif implementation == "amd":
-            devices = [ "ROCm" ]
+            devices = [ "cuda" ]
             frameworks = [ "pytorch" ]
+            execution_envs.remove("Docker")
 
         elif implementation == "neuralmagic":
             devices = [ "CUDA" ]