Add support for MI300X_A1

ROCm · Sep 11, 2024 · 5ca20b6 · 5ca20b6
1 parent 35e2f45
commit 5ca20b6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/omniperf_base.py b/src/omniperf_base.py
@@ -57,6 +57,7 @@
 
 MI300_CHIP_IDS = {
     "29856" : "MI300A_A1",
+    "29857" : "MI300X_A1",
     "29858" : "MI308X",
 }
 

diff --git a/src/omniperf_soc/soc_base.py b/src/omniperf_soc/soc_base.py
@@ -203,6 +203,9 @@ def populate_mspec(self):
             # We need to distinguish MI308X by peeking reported num CUs
             elif self._mspec.cu_per_gpu == "80" or "MI308X" in self.check_arch_override():
                 self._mspec.gpu_model = "MI308X"
+            # Use Chip ID to distinguish the gpu model by using built-in dictionary
+            elif self._mspec.chip_id in MI300_CHIP_IDS:
+                self._mspec.chip_id = MI300_CHIP_IDS[self._mspec.chip_id]
             else:
                 console_error(
                     "Cannot parse MI300 details from rocminfo. Please verify output or set the arch using (e.g.,) "