From dbb7f4d493bb0a65231564425b0d7812e2538edd Mon Sep 17 00:00:00 2001 From: vedithal-amd Date: Tue, 6 May 2025 18:23:08 -0400 Subject: [PATCH] Use gpu model series instead of gpu model name for testing (#696) --- tests/test_TCP_counters.py | 11 ++++++----- tests/test_profile_general.py | 7 ++++--- 2 files changed, 10 insertions(+), 8 deletions(-) diff --git a/tests/test_TCP_counters.py b/tests/test_TCP_counters.py index c304a9b48f..ab002da99a 100644 --- a/tests/test_TCP_counters.py +++ b/tests/test_TCP_counters.py @@ -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 diff --git a/tests/test_profile_general.py b/tests/test_profile_general.py index fdb06acd23..34c40d2c2a 100644 --- a/tests/test_profile_general.py +++ b/tests/test_profile_general.py @@ -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