Skip to content

Commit

Permalink
Use MI300 chip_id instead of model to detect XCD count (#448)
Browse files Browse the repository at this point in the history
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>
  • Loading branch information
benrichard-amd authored Oct 16, 2024
1 parent e633d17 commit a236fe0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/omniperf_soc/soc_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -187,7 +187,7 @@ def populate_mspec(self):
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
Expand Down

0 comments on commit a236fe0

Please sign in to comment.