Fix benchmark device arg

Signed-off-by: aviator19941 <avinash.sharma@amd.com>
nod-ai · Nov 4, 2024 · 31b59a8 · 31b59a8
1 parent 1ab391d
commit 31b59a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sharktank/sharktank/utils/export_artifacts.py b/sharktank/sharktank/utils/export_artifacts.py
@@ -256,7 +256,7 @@ def iree_benchmark_vmfb(
                 for i in range(self.tensor_parallelism_size)
             ]
             devices = [
-                f"--device=hip[{i}]" for i in range(self.tensor_parallelism_size)
+                f"--device=hip://{i}" for i in range(self.tensor_parallelism_size)
             ]
         else:
             params = [f"--parameters=model={irpa_path}"]