Fix typo in default_rocm_arch (#1998)

nod-ai · Nov 30, 2023 · 78c607e · 78c607e
1 parent 666e601
commit 78c607e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shark/iree_utils/gpu_utils.py b/shark/iree_utils/gpu_utils.py
@@ -104,7 +104,7 @@ def get_devices_info_from_dump(dump):
         print(f"Found ROCm device arch : {arch_in_device_dump}")
         return arch_in_device_dump
 
-    default_rocm_arch = "gfx_1100"
+    default_rocm_arch = "gfx1100"
     print(
         "Did not find ROCm architecture from `--iree-rocm-target-chip` flag"
         "\n or from `iree-run-module --dump_devices=rocm` command."