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>
[ROCm/rocprofiler-compute commit: a236fe0412]
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user