PR #8471 [LoweringContext] SPMD propagation

[LoweringContext] SPMD propagation #8471

tengyifei merged 5 commits into pytorch:master from rpsilva-aws:rpsilva_spmd_lc_v2

[LoweringContext] SPMD propagation

b18ba993

Improve IR dump graph test cleanups

9a98a366

rpsilva-aws marked this pull request as ready for review 1 year ago

jeffhataws requested a review from

tengyifei 1 year ago

jeffhataws requested a review from

ManfeiBai 1 year ago

jeffhataws requested a review from

miladm 1 year ago

tengyifei added tpuci

tengyifei commented on 2024-12-09

tengyifei approved these changes on 2024-12-09

Fix the setup for SPMD graph dump test

c48ea1f4

jeffhataws approved these changes on 2024-12-09

tengyifei approved these changes on 2024-12-09

Introduce run_test utils to accommodate test/tpu

0b7a20e1

[AdHoc] Add inline comment for BuildForiLoop

20debc49

rpsilva-aws force pushed from 14c61950 to 20debc49 1 year ago

tengyifei merged b068cab9 into master 1 year ago

rpsilva-aws deleted the rpsilva_spmd_lc_v2 branch 1 year ago

Reviewers

tengyifei

jeffhataws

ManfeiBai

miladm

Assignees

No one assigned

Labels

None yet

Milestone

No milestone