Patch specs enhancement for Mi300

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


[ROCm/rocprofiler-compute commit: f49d7f7f0c]
Tento commit je obsažen v:
colramos-amd
2024-01-24 12:58:13 -06:00
odevzdal Cole Ramos
rodič ee21c4d5ba
revize dcdd64b6bd
+1 -1
Zobrazit soubor
@@ -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):