Skip to content

Pull requests: openxla/shardy

New pull request New
65 Open 1,256 Closed
65 Open 1,256 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 copybara by using bzlmod for zlib and zstd
#1420 opened Jun 19, 2026 by copybara-service Bot Loading…
@Varcho
Fix out-of-bounds crash in distributeAxesAcrossFactors
#1418 opened Jun 18, 2026 by copybara-service Bot Loading…
@bixia1
[SDY] enable stablehlo.reverse interpreter test.
#1416 opened Jun 16, 2026 by copybara-service Bot Loading…
@Varcho
[SDY][PadForDivisibility] Move getEvenlySharded to utils.
#1415 opened Jun 16, 2026 by copybara-service Bot Loading…
@Varcho
[SDY][PadForDivisibility] Handle sdy.all_to_all in pad-for-divisibility. This involves (potentially) padding before the collective op and trimming afterwards.
#1414 opened Jun 16, 2026 by copybara-service Bot Loading…
@Varcho
[StableHLO][OpenXLA] Fix interpreter self-loop deadlock in collective_permute and re-enable shardy test
#1413 opened Jun 13, 2026 by copybara-service Bot Loading…
@srishti-pm
[SDY][Waymo-Bug] Add SplitReshardingDimensionsPass before ReshardToCollectives.
#1411 opened Jun 13, 2026 by copybara-service Bot Loading…
@Varcho
Rewrite ManualComputationOp shardings to use fragment local mesh in MPMD export.
#1408 opened Jun 11, 2026 by copybara-service Bot Loading…
@bixia1
Reverts 266760d0aadc4d355a9f7738229ecaa87dcb9f49
#1406 opened Jun 10, 2026 by copybara-service Bot Loading…
[consts] Fix handling of closed-over constants in MPMD
#1404 opened Jun 9, 2026 by copybara-service Bot Loading…
Add option xla_shardy_options='block_reversed_dims:true' for blocking sharding
#1400 opened Jun 8, 2026 by copybara-service Bot Loading…
@bixia1
Extend the canonicalizer to also inline manual_computation with equal global and
#1396 opened Jun 5, 2026 by copybara-service Bot Loading…
@bixia1
[mpmd] Prefer producer fragment when merging inferred fragments to avoid false dependencies.
#1393 opened Jun 4, 2026 by copybara-service Bot Loading…
[mpmd] Restrict MergeFragments to adjacent fragments.
#1389 opened Jun 2, 2026 by copybara-service Bot Loading…
[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] 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
ProTip! Adding no:label will show everything without a label.