Detection of MI308X and hardcode mclk to address bug in rocm-smi

Signed-off-by: coleramos425 <colramos@amd.com>
This commit is contained in:
coleramos425
2024-06-03 17:07:23 +00:00
committed by Cole Ramos
parent 295b344646
commit 1d19ae9483
2 changed files with 11 additions and 0 deletions
+6
View File
@@ -194,6 +194,12 @@ class OmniSoC_Base:
or "MI300X" in self.check_arch_override()
):
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()
):
self._mspec.gpu_model = "MI308X"
else:
console_error(
"Cannot parse MI300 details from rocminfo. Please verify output or set the arch using (e.g.,) "
+5
View File
@@ -69,6 +69,11 @@ class gfx942_soc(OmniSoC_Base):
)
# self.roofline_obj = Roofline(args, self._mspec)
# --showmclkrange is broken in MI308X, hardcode freq
if self._mspec.gpu_model == "MI308X":
self._mspec.max_mclk = 1300
self._mspec.cur_mclk = 1300
# Set arch specific specs
self._mspec._l2_banks = 16
self._mspec.lds_banks_per_cu = 32