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

Signed-off-by: coleramos425 <colramos@amd.com>
Этот коммит содержится в:
coleramos425
2024-06-03 17:07:23 +00:00
коммит произвёл Cole Ramos
родитель 295b344646
Коммит 1d19ae9483
2 изменённых файлов: 11 добавлений и 0 удалений
+6
Просмотреть файл
@@ -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
Просмотреть файл
@@ -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