Skip to content

Commit

Permalink
Use MI300 chip_id instead of model to detect XCD count (ROCm#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>
Signed-off-by: xuchen-amd <xuchen@amd.com>
  • Loading branch information
benrichard-amd authored and xuchen-amd committed Oct 21, 2024
1 parent 403f921 commit 0f37a5c
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 0f37a5c

Please sign in to comment.