AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
vedithal-amd
2025-05-06 18:23:08 -04:00
committed by GitHub
parent abd500593b
commit dbb7f4d493
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_TCP_counters.py
+6 -5
View File
@@ -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
tests/test_profile_general.py
+4 -3
View File
@@ -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