Skip to content

Commit

Permalink
apply formatting
Browse files Browse the repository at this point in the history
Signed-off-by: Karl W Schulz <karl.schulz@amd.com>
  • Loading branch information
koomie committed Mar 12, 2024
1 parent 455645b commit 458087b
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions src/omniperf_soc/soc_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 458087b

Please sign in to comment.