Skip to content

Pull requests: openxla/shardy

New pull request New
52 Open 1,225 Closed
52 Open 1,225 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

Add ExplicitGatherBatchingPattern pass in Shardy .
#1376 opened May 30, 2026 by copybara-service Bot Loading…
[mpmd] Add fragment query, positioning, and attribute save/restore utilities with tests.
#1375 opened May 29, 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…
Add a check for communication-free slice dimensions in PadForDivisibility.
#1370 opened May 29, 2026 by copybara-service Bot Loading…
@bixia1
[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…
[consts] Adapt MPMD code to the new JAX handling of closed-over constants
#1115 opened Apr 7, 2026 by copybara-service Bot Loading…
Place uniquified return fragments adjacent to their producer
#1088 opened Mar 26, 2026 by copybara-service Bot Loading…
Fix Gather/Scatter sharding rule for dimensions not in start_index_map.
#1085 opened Mar 25, 2026 by copybara-service Bot Loading…
@ZixuanJiang
ProTip! Follow long discussions with comments:>50.