PR #7604 Handle the parameter wrapping for SPMD

Handle the parameter wrapping for SPMD #7604

JackCaoG merged 2 commits into master from JackCaoG/auto_wrap_spmd

Handle the parameter wrapping for SPMD

726ff54e

add test

c30cacd1

JackCaoG added tpuci

JackCaoG added backport_2.4

alanwaketan requested a review from

alanwaketan 1 year ago

JackCaoG marked this pull request as ready for review 1 year ago

alanwaketan commented on 2024-07-02

alanwaketan approved these changes on 2024-07-02

JackCaoG commented on 2024-07-02

JackCaoG merged 4b7e518d into master 1 year ago

Reviewers

alanwaketan

Assignees

No one assigned

Labels

backport_2.4

Milestone

No milestone