Skip to content

Pull requests: openxla/shardy

New pull request New
55 Open 1,228 Closed
55 Open 1,228 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

[mpmd] Merge inferred fragments inline in UniquifyFunctionInputsOutputsPass.
#1382 opened Jun 1, 2026 by copybara-service Bot Loading…
[mpmd] Add utils for finding and moving fragments during fragment merging.
#1381 opened Jun 1, 2026 by copybara-service Bot Loading…
[mpmd] Move MergeAttributes to utils and add unit tests.
#1380 opened Jun 1, 2026 by copybara-service Bot Loading…
[mpmd] Make MergeRegionOps internal to utils.
#1379 opened Jun 1, 2026 by copybara-service Bot Loading…
[mpmd] Add MergeFragments utility to simplify fragment merging.
#1378 opened Jun 1, 2026 by copybara-service Bot Loading…
[mpmd] Update uniquify test to chain merge-inferred-fragments pass.
#1377 opened Jun 1, 2026 by copybara-service Bot Loading…
[mpmd] Refactor UniquifyFunctionInputsOutputsPass to use shared utilities.
#1374 opened May 29, 2026 by copybara-service Bot Loading…
[mpmd] Simplify inline merge: always merge into last same-mesh fragment.
#1373 opened May 29, 2026 by copybara-service Bot Loading…
[mpmd] Fix attribute preservation and add tests for inline merge in UniquifyFunctionInputsOutputsPass.
#1369 opened May 29, 2026 by copybara-service Bot Loading…
Ignore intra-mesh broadcasts, as they are trivial ops.
#1363 opened May 21, 2026 by copybara-service Bot Loading…
[SDY:MPMD] Remove MIN_JAXLIB_VERSION_FOR_SCHEDULE_RULES
#1362 opened May 21, 2026 by copybara-service Bot Loading…
Inline producer-consumer merging into UniquifyFunctionInputsOutputsPass.
#1339 opened May 13, 2026 by copybara-service Bot Loading…
[mpmd] Fix visibility of overridden methods.
#1336 opened May 12, 2026 by copybara-service Bot Loading…
Add missing loads for Bazel 8+ support
#1334 opened May 12, 2026 by cerisier Loading…
2
[XLA] Support xla.debug.Log in sharding passes
#1333 opened May 11, 2026 by copybara-service Bot Loading…
[mpmd] Extract WrapBlockArgReturnsPass from UniquifyFunctionInputsOutputsPass.
#1302 opened May 5, 2026 by copybara-service Bot Loading…
[ReplicaGroupV3][JAX+stablehlo][6/n] Emit RGV3 from JAX in shard_map
#1287 opened May 1, 2026 by copybara-service Bot Loading…
@Varcho
[mpmd] Merge inferred fragments inline in UniquifyFunctionInputsOutputsPass.
#1274 opened Apr 30, 2026 by copybara-service Bot Loading…
Pass tokens unmodified through sdy-basic-propagate.
#1259 opened Apr 30, 2026 by copybara-service Bot Loading…
[mpmd] Add UniquifyAndMergeReturnsV2Pass: create-then-merge approach.
#1258 opened Apr 29, 2026 by copybara-service Bot Loading…
Add UniquifyAndMergeReturnsPass to merge return copies into existing fragments.
#1256 opened Apr 29, 2026 by copybara-service Bot Loading…
[SDY] Update JAX export and PartIR to provide a target SDY version
#1219 opened Apr 23, 2026 by copybara-service Bot Loading…
[mpmd] Merge sideways in export pipeline.
#1198 opened Apr 16, 2026 by copybara-service Bot Loading…
Avoid usage of deprecated jax.core APIs.
#1189 opened Apr 15, 2026 by copybara-service Bot Loading…
Remove mesh deduplication in LiftInlinedMeshesPass.
#1138 opened Apr 8, 2026 by copybara-service Bot Loading…
ProTip! What’s not been updated in a month: updated:<2026-05-01.