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