Patch specs enhancement for Mi300

Signed-off-by: colramos-amd <colramos@amd.com>
ROCm · Jan 24, 2024 · f49d7f7 · f49d7f7
1 parent e5ac5cc
commit f49d7f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils/specs.py b/src/utils/specs.py
@@ -202,7 +202,7 @@ def gpuinfo():
 
     # specify gpu name for gfx942 hardware
     if gpu_name == "MI300":
-        gpu_name = list(SUPPORTED_ARCHS[gpu_arch].values())[0]
+        gpu_name = list(SUPPORTED_ARCHS[gpu_arch].values())[0][0]
     if (gpu_info['gpu_arch'] == "gfx942") and ("MI300A" in rocminfo_full):
         gpu_name = "MI300A_A1"
     if (gpu_arch == "gfx942") and ("MI300A" not in rocminfo_full):