xla
Transfer shards from server using replication
#4508

Merged

Transfer shards from server using replication #4508

yeounoh merged 5 commits into master from spmd_preserve_sharding

yeounoh added distributed

yeounoh requested a review from

JackCaoG 2 years ago

yeounoh requested a review from

steventk-g 2 years ago

yeounoh assigned

yeounoh 2 years ago

yeounoh force pushed from c4842503 to 49f21e18 2 years ago

yeounoh force pushed from 49f21e18 to ed77f6aa 2 years ago

JackCaoG commented on 2023-01-28

yeounoh force pushed from 2622cfc8 to 1d447bd0 2 years ago

yeounoh force pushed from 1d447bd0 to eeb88a05 2 years ago

yeounoh force pushed from eeb88a05 to a286d6ea 2 years ago

yeounoh requested a review from

will-cromar 2 years ago

[SPMD] refactor & add more debugging probes

5b01ccab

[SPMD] Transfer shards back to host with replication

c7fa79bd

Remove debugging probes after debugging.

9f03dd9a

[SPMD] Add test_transfer_sharded_data_to_host test case.

88eba658

yeounoh force pushed from a286d6ea to 556cc221 2 years ago

Refactoring

48736237

yeounoh force pushed from 556cc221 to 48736237 2 years ago

JackCaoG approved these changes on 2023-02-01

yeounoh merged 780a8231 into master 2 years ago

Reviewers

JackCaoG

steventk-g

will-cromar

Assignees

yeounoh

Labels

distributed

Milestone

No milestone