Skip to content

[scan] Add a test under SPMD#8419

Merged
tengyifei merged 1 commit intomasterfrom
yifeit/scan-spmd
Dec 4, 2024
Merged

[scan] Add a test under SPMD#8419
tengyifei merged 1 commit intomasterfrom
yifeit/scan-spmd

Conversation

@tengyifei
Copy link
Copy Markdown
Collaborator

Verifies that the GSPMD sharding annotation propagation pass can propagate through a While op and through the Body computation just fine.

@tengyifei tengyifei force-pushed the yifeit/scan-spmd branch 2 times, most recently from b3f1ff5 to c9a5907 Compare November 26, 2024 22:07
@JackCaoG
Copy link
Copy Markdown
Collaborator

@yaochengji FYI

@tengyifei tengyifei requested a review from JackCaoG November 26, 2024 23:09
@JackCaoG
Copy link
Copy Markdown
Collaborator

I will take a look today.

Comment thread test/scan/test_scan_spmd.py Outdated
Comment thread test/scan/test_scan_spmd.py Outdated
Comment thread test/scan/test_scan_spmd.py Outdated
Verifies that the GSPMD sharding annotation propagation pass can
propagate through a While op and through the Body computation just fine.
@tengyifei tengyifei merged commit 2ba14fc into master Dec 4, 2024
rpsilva-aws pushed a commit to rpsilva-aws/xla that referenced this pull request Dec 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants