search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.73k stars 438 forks source link

PR #19775: [PJRT:GPU] Fix device numbering in topology creation #19776

Closed copybara-service[bot] closed 23 hours ago

copybara-service[bot] commented 1 day ago

PR #19775: [PJRT:GPU] Fix device numbering in topology creation

Imported from GitHub PR https://github.com/openxla/xla/pull/19775

Copybara import of the project:

-- 16710c507e107da3b09edf9039ceb0f3cf4542d2 by Jaroslav Sevcik jsevcik@nvidia.com:

[PJRT:GPU] Fix device numbering in topology creation

Merging this change closes #19775

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/19775 from jaro-sevcik:get-topology-desc-device-id-fix 16710c507e107da3b09edf9039ceb0f3cf4542d2