Skip to content

Pull requests: openxla/shardy

New pull request New
43 Open 759 Closed
43 Open 759 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

Move macros.h from shardy/dialect/sdy/transforms/common to shardy/dialect/sdy/ir.
#813 opened Oct 3, 2025 by copybara-service bot Loading…
@ZixuanJiang
Update findCommonAxesOnUnaryOperation in explicit reshards.
#812 opened Oct 3, 2025 by copybara-service bot Loading…
@ZixuanJiang
Move getPrefixWithoutOverlap from basic_factor_propagation.cc to ir/utils.h.
#811 opened Oct 3, 2025 by copybara-service bot Loading…
@ZixuanJiang
Transfer instead of adding a fragment
#810 opened Oct 3, 2025 by copybara-service bot Loading…
[mpmd] Pass cloneInferredFragments to post-inline MergeInferredFragmentsPass
#809 opened Oct 3, 2025 by copybara-service bot Loading…
Keep reshards that has non-equivalent input and output meshes.
#804 opened Oct 1, 2025 by copybara-service bot Loading…
[mpmd] Support heterogeneous meshes.
#801 opened Oct 1, 2025 by copybara-service bot Loading…
[mpmd] Add custom pipeline API for user-defined scheduling and merging
#798 opened Sep 30, 2025 by copybara-service bot Loading…
[mpmd] Use stage id instead of mesh id in pipeline_schedule.cc
#797 opened Sep 30, 2025 by copybara-service bot Loading…
Early exit before all reduces if op has no results also on the full version.
#792 opened Sep 30, 2025 by copybara-service bot Loading…
Confine logics to find reduction axes and to check for unreduced axes inside the method that inserts all reduces.
#787 opened Sep 29, 2025 by copybara-service bot Loading…
Move logic related to allreduce out of process op method.
#785 opened Sep 29, 2025 by copybara-service bot Loading…
fix relayout propagation for MPMD.
#783 opened Sep 29, 2025 by copybara-service bot Loading…
@liepieshov
Unify flows for finding common axes and building reduction axes out of it.
#780 opened Sep 25, 2025 by copybara-service bot Loading…
Add canonicalization for redundant reshard ops.
#747 opened Sep 16, 2025 by copybara-service bot Loading…
Hard fail if target has unreduced axes that are not contained in the source unreduced axes. This case for unreduced axes is not supported.
#742 opened Sep 15, 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
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…
Reshard to collectives also for the default/minimal version of shardy partitioner.
#687 opened Aug 27, 2025 by copybara-service bot Loading…
[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…
ProTip! Add no:assignee to see everything that’s not assigned.