From dcdd64b6bdf2f19c15c2d170335159eae15b711a Mon Sep 17 00:00:00 2001 From: colramos-amd Date: Wed, 24 Jan 2024 12:58:13 -0600 Subject: [PATCH] Patch specs enhancement for Mi300 Signed-off-by: colramos-amd [ROCm/rocprofiler-compute commit: f49d7f7f0cf4193d14519a582caabafc7da0931e] --- projects/rocprofiler-compute/src/utils/specs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/projects/rocprofiler-compute/src/utils/specs.py b/projects/rocprofiler-compute/src/utils/specs.py index ff4897bee7..750e9dbf84 100644 --- a/projects/rocprofiler-compute/src/utils/specs.py +++ b/projects/rocprofiler-compute/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):