xla
Fix the error where mark_step does not materalize tensors on SPMD:0
#5281

Merged

Fix the error where mark_step does not materalize tensors on SPMD:0 #5281

JackCaoG merged 3 commits into master from JackCaoG/fix_mark_step_on_spmd_device

JackCaoG requested a review from

alanwaketan 2 years ago

alanwaketan commented on 2023-07-05

alanwaketan approved these changes on 2023-07-05

alanwaketan commented on 2023-07-05

Fix the error where mark_step does not materalize tensors on SPMD:0

36310efe

typo

54c8b46c

JackCaoG force pushed from 99142312 to 54c8b46c 2 years ago

fix test_non_tensor_scalar

f8755503

JackCaoG merged cb5944ae into master 2 years ago

Reviewers

alanwaketan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone