Use MI300 chip_id instead of model to detect XCD count (#448)
In a previous change we started using "MI300" for gpu_model instead of the full "MI300X_A0" or "MI300X_A1", etc. The XCD detection code was using gpu_model and expecting the full name, causing the XCD count = 1. Passing chip_id fixes the issue. Signed-off-by: benrichard-amd <ben.richard@amd.com>
Этот коммит содержится в:
@@ -187,7 +187,7 @@ class OmniSoC_Base:
|
||||
self._mspec.chip_id = MI300_CHIP_IDS[self._mspec.chip_id]
|
||||
|
||||
self._mspec.num_xcd = str(
|
||||
total_xcds(self._mspec.gpu_model, self._mspec.compute_partition)
|
||||
total_xcds(self._mspec.chip_id, self._mspec.compute_partition)
|
||||
)
|
||||
|
||||
@demarcate
|
||||
|
||||
Ссылка в новой задаче
Block a user