Comply to formatting

Signed-off-by: coleramos425 <colramos@amd.com>


[ROCm/rocprofiler-compute commit: 7046ea15bd]
This commit is contained in:
coleramos425
2024-06-03 17:19:57 +00:00
committed by Cole Ramos
parent 67c4075950
commit 68387da116
2 changed files with 5 additions and 5 deletions
@@ -195,10 +195,7 @@ class OmniSoC_Base:
):
self._mspec.gpu_model = "MI300X_A1"
# We need to distinguish MI308X by peeking reported num CUs
elif(
self._mspec.cu_per_gpu == "80"
or "MI308X" in self.check_arch_override()
):
elif self._mspec.cu_per_gpu == "80" or "MI308X" in self.check_arch_override():
self._mspec.gpu_model = "MI308X"
else:
console_error(
@@ -562,7 +562,10 @@ def total_xcds(archname, compute_partition):
mi300a_archs = ["mi300a_a0", "mi300a_a1"]
mi300x_archs = ["mi300x_a0", "mi300x_a1"]
mi308x_archs = ["mi308x"]
if archname.lower() in mi300a_archs + mi300x_archs + mi308x_archs and compute_partition == "NA":
if (
archname.lower() in mi300a_archs + mi300x_archs + mi308x_archs
and compute_partition == "NA"
):
console_error("Invalid compute partition found for {}".format(archname))
if archname.lower() not in mi300a_archs + mi300x_archs + mi308x_archs:
return 1