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

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
 

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