xla
[LoweringContext] Support an optimized parameter mapping for SPMD
#8460

Merged

[LoweringContext] Support an optimized parameter mapping for SPMD #8460

tengyifei merged 1 commit into pytorch:master from rpsilva-aws:rpsilva_lc_mapping_v3

tengyifei requested a review from

tengyifei 1 year ago

tengyifei added tpuci

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

tengyifei approved these changes on 2024-12-05

[LoweringContext] Support an optimized parameter mapping for SPMD

9858577e

rpsilva-aws force pushed from 8fd7ac76 to 9858577e 1 year ago

tengyifei merged 5d11f660 into master 1 year ago

rpsilva-aws deleted the rpsilva_lc_mapping_v3 branch 1 year ago

Reviewers

tengyifei

Assignees

No one assigned

Labels

None yet

Milestone

No milestone