Patch specs enhancement for Mi300

Signed-off-by: colramos-amd <colramos@amd.com>


[ROCm/rocprofiler-compute commit: f49d7f7f0c]
このコミットが含まれているのは:
colramos-amd
2024-01-24 12:58:13 -06:00
committed by Cole Ramos
コミット dcdd64b6bd
+1 -1
ファイルの表示
@@ -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):