xla
[SPMD][DTensor] introduce xla_distribute_module for DTensor integration
#6683

Merged

[SPMD][DTensor] introduce xla_distribute_module for DTensor integration #6683

yeounoh merged 1 commit into master from xla_distribute_module

yeounoh added distributed

yeounoh assigned

yeounoh 2 years ago

yeounoh force pushed from c88e1897 to a157894b 2 years ago

yeounoh requested a review from

alanwaketan 2 years ago

yeounoh requested a review from

wanchaol 2 years ago

yeounoh force pushed from a157894b to 26fc3a84 2 years ago

yeounoh force pushed from 26fc3a84 to 30850e16 2 years ago

yeounoh requested a review from

JackCaoG 2 years ago

yeounoh force pushed from 30850e16 to 12b2f5f0 2 years ago

yeounoh force pushed from 12b2f5f0 to 3ab5d64a 2 years ago

yeounoh force pushed from 3ab5d64a to 4caa123b 2 years ago

alanwaketan approved these changes on 2024-03-07

[SPMD][DTensor] introduce xla_distribute_module for DTensor integration

e659a762

yeounoh force pushed from 4caa123b to e659a762 2 years ago

yeounoh merged b6b9c6da into master 2 years ago

Reviewers

alanwaketan

wanchaol

JackCaoG

Assignees

yeounoh

Labels

distributed

Milestone

No milestone