Skip to content

Pull requests: openxla/shardy

New pull request New
59 Open 1,079 Closed
59 Open 1,079 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 an executable test to check the conversion of sdy.reduce_scatter in convert-global-to-local.
#1158 opened Apr 10, 2026 by copybara-service bot Loading…
@bixia1
Allow type conversion for ops from unknown dialects.
#1157 opened Apr 10, 2026 by copybara-service bot Loading…
@bixia1
[consts] Adapt MPMD code to the new JAX handling of closed-over constants
#1156 opened Apr 10, 2026 by copybara-service bot Loading…
Push shardy outliner up out-of shardy partitioner pipeline.
#1154 opened Apr 9, 2026 by copybara-service bot Loading…
@ekayaaslan
[mpmd] Remove kIsSdyLowered attribute.
#1149 opened Apr 9, 2026 by copybara-service bot Loading…
[mpmd] Remove deprecated attributes.
#1143 opened Apr 8, 2026 by copybara-service bot Loading…
Remove mesh deduplication in LiftInlinedMeshesPass.
#1138 opened Apr 8, 2026 by copybara-service bot Loading…
Fix the handling of sharded reduction factors in Scatter op.
#1136 opened Apr 7, 2026 by copybara-service bot Loading…
@bixia1
Push shardy outliner up past InsertExplicitReshards pass.
#1132 opened Apr 7, 2026 by copybara-service bot Loading…
@ekayaaslan
Push shardy outliner up past canonicalizer on reshards.
#1131 opened Apr 7, 2026 by copybara-service bot Loading…
@ekayaaslan
Push shardy outliner up past ReshardToCollectives pass.
#1130 opened Apr 7, 2026 by copybara-service bot Loading…
@ekayaaslan
Push shardy outliner up past RemoveAllGatherReduceScatterForCMV1 pass.
#1129 opened Apr 7, 2026 by copybara-service bot Loading…
@ekayaaslan
Push shardy outliner up saving module after shardy partitioner.
#1128 opened Apr 7, 2026 by copybara-service bot Loading…
@ekayaaslan
Push shardy outliner up into shardy partitioner pipeline.
#1127 opened Apr 7, 2026 by copybara-service bot Loading…
@ekayaaslan
Push shardy outliner up RemovePropagationDebugInfo pass.
#1125 opened Apr 7, 2026 by copybara-service bot Loading…
@ekayaaslan
Push shardy inliner down past ManualAxesCleanup pass.
#1122 opened Apr 7, 2026 by copybara-service bot Loading…
@ekayaaslan
[consts] Adapt MPMD code to the new JAX handling of closed-over constants
#1115 opened Apr 7, 2026 by copybara-service bot Loading…
[ReplicaGroupV3][JAX+stablehlo] Shardy updates for RGV3
#1091 opened Mar 27, 2026 by copybara-service bot Loading…
@Varcho
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
[mpmd] Add flag-guarded sideways merge of inferred fragments in export pipeline for forward only programs.
#1082 opened Mar 25, 2026 by copybara-service bot Loading…
[mpmd] Detect backward deps in input/output transfers
#1072 opened Mar 19, 2026 by copybara-service bot Loading…
Prototype enabling HloShardingV3
#1041 opened Mar 10, 2026 by copybara-service bot Loading…
@KanishAnand
Fix compiling with -Werror
#1025 opened Mar 5, 2026 by nsmithtt Loading…
Lower jax.debug.print without using Python callbacks.
#1008 opened Feb 27, 2026 by copybara-service bot Loading…
ProTip! Exclude everything labeled bug with -label:bug.