Skip to content

Pull requests: openxla/shardy

New pull request New
41 Open 686 Closed
41 Open 686 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Fix SDY MPMD error handling
#738 opened Sep 12, 2025 by copybara-service bot Loading…
@JoelWee
[mpmd] Add rule-based scheduling pass
#737 opened Sep 12, 2025 by copybara-service bot Loading…
Add more unit tests for allreduces on unreduced axes.
#736 opened Sep 11, 2025 by copybara-service bot Loading…
Insert all reduces for unreduced axes also when the source unreduced axes is a strict subset of target unreduced axes.
#734 opened Sep 10, 2025 by copybara-service bot Loading…
[mpmd] Check SplitFragmentType and remat flag when verifying stage merging
#731 opened Sep 10, 2025 by copybara-service bot Loading…
Expand constant expressions on named computations with all constant like ops.
#728 opened Sep 9, 2025 by copybara-service bot Loading…
Refactor constant splitting to handle potentially constant named computations.
#727 opened Sep 9, 2025 by copybara-service bot Loading…
#sdy never hard fail in getSharding, only in setSharding.
#726 opened Sep 9, 2025 by copybara-service bot Loading…
@tomnatan30
Add shardy_mpmd extensions to jaxlib
#725 opened Sep 9, 2025 by copybara-service bot Loading…
@JoelWee
Introduce sdy-remove-all-gather-reduce-scatter-for-cmv1 pass.
#721 opened Sep 9, 2025 by copybara-service bot Loading…
@ZixuanJiang
Automated Code Change
#716 opened Sep 6, 2025 by copybara-service bot Loading…
[mpmd] Add mesh_name field to FragmentInfo
#710 opened Sep 4, 2025 by copybara-service bot Loading…
[mpmd] Move pre rule-based scheduling/merging passes to import pipeline
#709 opened Sep 4, 2025 by copybara-service bot Loading…
Apply add/remove has_side_effect in MPMD pipelines
#708 opened Sep 4, 2025 by copybara-service bot Loading…
Avoid CSE on stablehlo.custom_call marked with no_cse attribute
#707 opened Sep 4, 2025 by copybara-service bot Loading…
[mpmd] Use global mesh.
#700 opened Sep 2, 2025 by copybara-service bot Loading…
Insert all-reduces for unreduced axes only for the cases that the op has one result.
#691 opened Aug 28, 2025 by copybara-service bot Loading…
Reshard to collectives also for the default/minimal version of shardy partitioner.
#687 opened Aug 27, 2025 by copybara-service bot Loading…
[SDY] Add propagation edges for sharding constraints.
#658 opened Aug 7, 2025 by copybara-service bot Loading…
@Varcho
[SDY] Enable PropagationEdgesAttr verification for sdy.data_flow_edge operations.
#657 opened Aug 7, 2025 by copybara-service bot Loading…
@Varcho
[mpmd] Test different order
#654 opened Aug 6, 2025 by copybara-service bot Loading…
Verify that the shrdings inside the sdy.manual_computation shares the same mesh.
#624 opened Jul 23, 2025 by copybara-service bot Loading…
@ZixuanJiang
Add numerics check for unreduced e2e.
#571 opened Jun 6, 2025 by copybara-service bot Loading…
#sdy Mark unreduced ops with frontend attributes.
#567 opened Jun 5, 2025 by copybara-service bot Loading…
#sdy Add all-to-all fusion canonicalization pattern.
#516 opened May 7, 2025 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.