xla
Fix global_runtime_device_count() for spmd case
#6266

Merged

Fix global_runtime_device_count() for spmd case #6266

vanbasten23 merged 4 commits into master from fix_global_runtime_device_count_spmd

vanbasten23 force pushed from a58f3c6f to 84499bcc 2 years ago

vanbasten23 changed the title ~~Fix global_runtime_device_count for spmd case~~ Fix global_runtime_device_count() and local_device_count() for spmd case 2 years ago

vanbasten23 changed the title ~~Fix global_runtime_device_count() and local_device_count() for spmd case~~ Fix global_runtime_device_count() for spmd case 2 years ago

yeounoh commented on 2024-01-11

yeounoh approved these changes on 2024-01-11

vanbasten23 marked this pull request as ready for review 2 years ago

fix global_runtime_device_count for the spmd case

11e33f1a

unskip the test on cuda

d2e27d38

try to fix a spmd test

58d0f757

fix linter

4bf8d448

vanbasten23 force pushed from 9cfa8845 to 4bf8d448 2 years ago

vanbasten23 merged 27ae6dc9 into master 2 years ago

miladm assigned

vanbasten23 1 year ago

miladm added distributed

miladm added xla:gpu

Reviewers

yeounoh

Assignees

vanbasten23

Labels

xla:gpu distributed

Milestone

No milestone

xla Fix global_runtime_device_count() for spmd case #6266 Merged

Fix global_runtime_device_count() for spmd case #6266

xla
Fix global_runtime_device_count() for spmd case
#6266

Merged