fix formatting

Signed-off-by: Nick Curtis <nicholas.curtis@amd.com>


[ROCm/rocprofiler-compute commit: 5579beeed5]
Esse commit está contido em:
Nick Curtis authored and Cole Ramos committed 2024-05-28 15:34:00 -05:00
1 pai fbb384bd2f
commit 562e36a5c3
1 file changed
+11 -5
@@ -100,8 +100,8 @@ class OmniSoC_Base:
return [] return []
def check_arch_override(self): def check_arch_override(self):
if 'OMNIPERF_ARCH_OVERRIDE' in os.environ.keys(): if "OMNIPERF_ARCH_OVERRIDE" in os.environ.keys():
return os.environ['OMNIPERF_ARCH_OVERRIDE'] return os.environ["OMNIPERF_ARCH_OVERRIDE"]
@demarcate @demarcate
def populate_mspec(self): def populate_mspec(self):
@@ -183,14 +183,20 @@ class OmniSoC_Base:
0 0
][0] ][0]
if self._mspec.gpu_arch == "gfx942": if self._mspec.gpu_arch == "gfx942":
if "MI300A" in "\n".join(self._mspec._rocminfo) or 'MI300A' in self.check_arch_override(): if (
"MI300A" in "\n".join(self._mspec._rocminfo)
or "MI300A" in self.check_arch_override()
):
self._mspec.gpu_model = "MI300A_A1" self._mspec.gpu_model = "MI300A_A1"
elif "MI300X" in "\n".join(self._mspec._rocminfo) or 'MI300X' in self.check_arch_override(): elif (
"MI300X" in "\n".join(self._mspec._rocminfo)
or "MI300X" in self.check_arch_override()
):
self._mspec.gpu_model = "MI300X_A1" self._mspec.gpu_model = "MI300X_A1"
else: else:
console_error( console_error(
"Cannot parse MI300 details from rocminfo. Please verify output or set the arch using (e.g.,) " "Cannot parse MI300 details from rocminfo. Please verify output or set the arch using (e.g.,) "
"export OMNIPERF_ARCH_OVERRIDE=\"MI300A\"" 'export OMNIPERF_ARCH_OVERRIDE="MI300A"'
) )
self._mspec.num_xcd = str( self._mspec.num_xcd = str(