apply formatting

Signed-off-by: Karl W Schulz <[email protected]>
ROCm · Mar 12, 2024 · 458087b · 458087b
1 parent 455645b
commit 458087b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/omniperf_soc/soc_base.py b/src/omniperf_soc/soc_base.py
@@ -178,13 +178,15 @@ def populate_mspec(self):
             self._mspec.gpu_model = list(SUPPORTED_ARCHS[self._mspec.gpu_arch].values())[
                 0
             ][0]
-        if (self._mspec.gpu_arch == "gfx942"):
+        if self._mspec.gpu_arch == "gfx942":
             if "MI300A" in "\n".join(self._mspec._rocminfo):
                 self._mspec.gpu_model = "MI300A_A1"
-            elif "MI300X"  in "\n".join(self._mspec._rocminfo):
+            elif "MI300X" in "\n".join(self._mspec._rocminfo):
                 self._mspec.gpu_model = "MI300X_A1"
             else:
-                console_error("Cannot parse MI300 details from rocminfo. Please verify output.")
+                console_error(
+                    "Cannot parse MI300 details from rocminfo. Please verify output."
+                )
 
         self._mspec.num_xcd = str(
             total_xcds(self._mspec.gpu_model, self._mspec.compute_partition)