xla
Update test_xla_sharding.py to take into account the number of devices
#8001

Merged

Update test_xla_sharding.py to take into account the number of devices #8001

bhavya01 merged 2 commits into master from bhavya01-patch-16

Update test_xla_sharding.py to take into account the number of devices

1e4e5d23

bhavya01 requested a review from

zpcore 1 year ago

fix lint

973f9574

bhavya01 requested a review from

qihqi 1 year ago

bhavya01 assigned

bhavya01 1 year ago

zpcore approved these changes on 2024-09-12

bhavya01 merged fc03aef3 into master 1 year ago

Reviewers

zpcore

qihqi

Assignees

bhavya01

Labels

None yet

Milestone

No milestone