Use gpu model series instead of gpu model name for testing (#696)

[ROCm/rocprofiler-compute commit: dbb7f4d493]
Этот коммит содержится в:
vedithal-amd
2025-05-06 18:23:08 -04:00
коммит произвёл GitHub
родитель bd04d43108
Коммит 98367932a3
2 изменённых файлов: 10 добавлений и 8 удалений
+6 -5
Просмотреть файл
@@ -69,11 +69,12 @@ def gpu_soc():
## 3) Deduce gpu model name from arch
gpu_model = list(SUPPORTED_ARCHS[gpu_arch].keys())[0].upper()
if gpu_model == "MI300":
if chip_id in MI300_CHIP_IDS:
gpu_model = MI300_CHIP_IDS[chip_id]
else:
return None
# For testing purposes we only care about gpu model series not the specific model
# if gpu_model == "MI300":
# if chip_id in MI300_CHIP_IDS:
# gpu_model = MI300_CHIP_IDS[chip_id]
# else:
# return None
return gpu_model
+4 -3
Просмотреть файл
@@ -312,9 +312,10 @@ def gpu_soc():
## 3) Deduce gpu model name from arch
gpu_model = list(SUPPORTED_ARCHS[gpu_arch].keys())[0].upper()
if gpu_model not in ("MI50", "MI100", "MI200"):
if chip_id in CHIP_IDS:
gpu_model = CHIP_IDS[chip_id]
# For testing purposes we only care about gpu model series not the specific model
# if gpu_model not in ("MI50", "MI100", "MI200"):
# if chip_id in CHIP_IDS:
# gpu_model = CHIP_IDS[chip_id]
return gpu_model