Skip to content

Pull requests: openxla/shardy

New pull request New
54 Open 1,208 Closed
54 Open 1,208 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 shardy OSS build: add missing llvm_zlib and llvm_zstd deps to WORKSPACE
#1361 opened May 20, 2026 by copybara-service Bot Loading…
[mpmd] Bypass max_clones limit for trivial ops in mesh inference.
#1360 opened May 20, 2026 by copybara-service Bot Loading…
[SDY][PadForDivisibility] Add DotGeneralOpPattern to enforce kZero padding
#1359 opened May 20, 2026 by copybara-service Bot Loading…
@Varcho
Add conversion patterns to convert-global-to-local for stablehlo.reduce_window and stablehlo.select_and_scatter.
#1358 opened May 19, 2026 by copybara-service Bot Loading…
@bixia1
Validate transformer params are not transferred between meshes
#1357 opened May 19, 2026 by copybara-service Bot Loading…
Drop unused enable_late_inlining option from Shardy.
#1348 opened May 14, 2026 by copybara-service Bot Loading…
@ekayaaslan
[mpmd] Fail if inferred fragments remain.
#1344 opened May 14, 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] UniquifyAndMergeReturnsPass
#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…
[SDY] Update IFRT HloProgram and IFRT IR Programs to use a target SDY version
#1218 opened Apr 23, 2026 by copybara-service Bot Loading…
[SDY] Update PjRt MLIR/HLO conversion API to accept an optional SDY target version during serialization
#1217 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…
[consts] Add lots of logging
#1156 opened Apr 10, 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…
ProTip! What’s not been updated in a month: updated:<2026-04-20.