From 7046ea15bd830c159a60141c7cbb43e606acd721 Mon Sep 17 00:00:00 2001 From: coleramos425 Date: Mon, 3 Jun 2024 17:19:57 +0000 Subject: [PATCH] Comply to formatting Signed-off-by: coleramos425 --- src/omniperf_soc/soc_base.py | 5 +---- src/utils/utils.py | 5 ++++- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/omniperf_soc/soc_base.py b/src/omniperf_soc/soc_base.py index d0b01cbfda..76e798439b 100644 --- a/src/omniperf_soc/soc_base.py +++ b/src/omniperf_soc/soc_base.py @@ -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( diff --git a/src/utils/utils.py b/src/utils/utils.py index 9869c562a0..b99d6a0b43 100644 --- a/src/utils/utils.py +++ b/src/utils/utils.py @@ -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