From 458087b425b49daf9be4956fc3eef1295d8137bf Mon Sep 17 00:00:00 2001 From: Karl W Schulz Date: Tue, 12 Mar 2024 09:09:46 -0500 Subject: [PATCH] apply formatting Signed-off-by: Karl W Schulz --- src/omniperf_soc/soc_base.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/omniperf_soc/soc_base.py b/src/omniperf_soc/soc_base.py index a576c4b78..1e007035a 100644 --- a/src/omniperf_soc/soc_base.py +++ b/src/omniperf_soc/soc_base.py @@ -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)