Skip to content

Pull requests: openxla/shardy

New pull request New
32 Open 652 Closed
32 Open 652 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

Refactor to simplify mesh extraction in getMeshOrDefault.
#694 opened Aug 29, 2025 by copybara-service bot Loading…
Return the default mesh in case all shardings are replicated.
#693 opened Aug 29, 2025 by copybara-service bot Loading…
Hard fail for unreduced axes if source and target both have unreduced axes but have different meshes.
#692 opened Aug 28, 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…
Refactor to create mesh only once and pass to methods of shardy partitioner.
#690 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…
Add canonicalizer pass and dump after reshards also for the default/minimal version.
#685 opened Aug 27, 2025 by copybara-service bot Loading…
#sdy Make sure the named scope of shard maps is preserved.
#672 opened Aug 20, 2025 by copybara-service bot Loading…
@bartchr808
Support factor types in sharding rules for custom partitioning.
#668 opened Aug 14, 2025 by copybara-service bot Loading…
@bixia1
[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…
Reverts 417dcc0d08153de3d533390db5258035e5869b54
#510 opened May 7, 2025 by copybara-service bot Loading…
@beckerhe
Test commit to ensure I haven't broken the config
#427 opened Mar 19, 2025 by copybara-service bot Loading…
@ddunl
#sdy add temporary pass in XLA SDY import to not allow meshes with different axis shapes.
#415 opened Mar 11, 2025 by copybara-service bot Loading…
@bartchr808
Don't allow sideways propagation between operands/results when propagation direction is forward/backwards.
#410 opened Mar 6, 2025 by copybara-service bot Loading…
@tomnatan30
Create dump directory if not exist.
#372 opened Feb 7, 2025 by yliu120 Loading…
Replace std::bind with std::bind_front for PropagationDirectionAlongFactor.
#363 opened Feb 5, 2025 by copybara-service bot Loading…
@ZixuanJiang
#sdy add sharding rules for ragged_all_to_all custom call
#338 opened Jan 23, 2025 by copybara-service bot Loading…
@ghpvnist
Fixed url encoding for MD.
#336 opened Jan 23, 2025 by copybara-service bot Loading…
Update to match upstream API change (NFC).
#282 opened Dec 23, 2024 by copybara-service bot Loading…
@jpienaar
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.