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>
Этот коммит содержится в:
Ben Richard
2024-10-16 15:41:11 -04:00
коммит произвёл GitHub
родитель e633d17372
Коммит a236fe0412
+1 -1
Просмотреть файл
@@ -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