apply formatting

Signed-off-by: Karl W Schulz <karl.schulz@amd.com>
Cette révision appartient à :
Karl W Schulz
2024-03-12 09:09:46 -05:00
Parent 455645b9c7
révision 458087b425
+5 -3
Voir le fichier
@@ -178,13 +178,15 @@ class OmniSoC_Base:
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)