xla
[SPMD] UseVirtualDevice() is set, when XLA_AUTO_SPMD=1
#6894

Merged

[SPMD] UseVirtualDevice() is set, when XLA_AUTO_SPMD=1 #6894

yeounoh merged 1 commit into master from improve_auto_sharding_ui

UseVirtualDevice() is set, when XLA_AUTO_SPMD=1

77097760

yeounoh added distributed

yeounoh requested a review from

JackCaoG 2 years ago

yeounoh assigned

yeounoh 2 years ago

JackCaoG approved these changes on 2024-04-05

alanwaketan approved these changes on 2024-04-05

yeounoh merged f44875c0 into master 2 years ago

Reviewers

alanwaketan

JackCaoG

Assignees

yeounoh

Labels

distributed

Milestone

No milestone