xla
[SPMD] fixes bugs with AssignIrValue & ExecuteReplicated
#4233

Merged

[SPMD] fixes bugs with AssignIrValue & ExecuteReplicated #4233

yeounoh merged 3 commits into master from spmd_mark_step

yeounoh added distributed

yeounoh requested a review from

jonb377 2 years ago

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 aa6a0da6 to 62f21792 2 years ago

JackCaoG commented on 2022-11-22

yeounoh force pushed from 94f94ba1 to b083799c 2 years ago

yeounoh force pushed from b083799c to 63773a9f 2 years ago

JackCaoG commented on 2022-11-23

JackCaoG approved these changes on 2022-11-23

yeounoh force pushed from 63773a9f to a16dc28a 2 years ago

[SPMD] fixes bugs with AssignIrValue & ExecuteReplicated

c8f70ad4

rebase master

55f161d3

Add/modify a unit test case, test_mark_step_with_sharding

f167e305

yeounoh force pushed from a16dc28a to f167e305 2 years ago

yeounoh merged 25cb9975 into master 2 years ago

Reviewers

JackCaoG

jonb377

steventk-g

Assignees

yeounoh

Labels

distributed

Milestone

No milestone