gecommit door
Cole Ramos
bovenliggende
1d19ae9483
commit
7046ea15bd
@@ -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(
|
||||
|
||||
@@ -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
|
||||
|
||||
Verwijs in nieuw issue
Block a user