Skip to content

Pull requests: openxla/shardy

New pull request New
42 Open 1,008 Closed
42 Open 1,008 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

Implement the conversion pattern for stablehlo.gather.
#1068 opened Mar 18, 2026 by copybara-service bot Loading…
@bixia1
Implement the conversion pattern for stablehlo.slice.
#1067 opened Mar 18, 2026 by copybara-service bot Loading…
@bixia1
[Shardy MPMD] Add support for unreduced axes in MPMD sharding.
#1062 opened Mar 17, 2026 by copybara-service bot Loading…
@emilyfertig
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…
Chain MPMD-introduced broadcasts in introduce_transfers.
#1000 opened Feb 24, 2026 by copybara-service bot Loading…
[mpmd] Add hashing support for PipelineSchedule
#982 opened Feb 18, 2026 by copybara-service bot Loading…
Add RemoveSubAxesInInputOutputShardingsPass to the export pipeline.
#979 opened Feb 13, 2026 by copybara-service bot Loading…
@ZixuanJiang
Make global meshes match local meshes.
#977 opened Feb 12, 2026 by copybara-service bot Loading…
Clean-up import_pipeline test.
#941 opened Jan 6, 2026 by copybara-service bot Loading…
rm unused function insertSeenValue
#934 opened Dec 18, 2025 by nsmithtt Loading…
1
Recover communication cost in explicit reshards.
#931 opened Dec 18, 2025 by copybara-service bot Loading…
@ZixuanJiang
Do not propagate shardings along reversed dimensions in Shardy.
#905 opened Dec 4, 2025 by copybara-service bot Loading…
@ZixuanJiang
Add support for custom sharding rules via ShardingRuleOpInterface
#885 opened Nov 14, 2025 by sshonTT Loading…
1
Add sharding rule for stablehlo.batch_norm_inference op
#847 opened Oct 15, 2025 by sshonTT Loading…
4
Support unpinned host as a type of on host
#827 opened Oct 8, 2025 by copybara-service bot Loading…
Update schedule in op_priority_propagation.cc.
#814 opened Oct 4, 2025 by copybara-service bot Loading…
@ZixuanJiang
[mpmd] Use stage id instead of mesh id in pipeline_schedule.cc
#797 opened Sep 30, 2025 by copybara-service bot Loading…
#sdy never hard fail in getSharding, only in setSharding.
#726 opened Sep 9, 2025 by copybara-service bot Loading…
@tomnatan30
Apply add/remove has_side_effect in MPMD pipelines
#708 opened Sep 4, 2025 by copybara-service bot Loading…
Avoid CSE on stablehlo.custom_call marked with no_cse attribute
#707 opened Sep 4, 2025 by copybara-service bot Loading…
[mpmd] Use global mesh.
#700 opened Sep 2, 2025 by copybara-service bot Loading…
Verify that the shrdings inside the sdy.manual_computation shares the same mesh.
#624 opened Jul 23, 2025 by copybara-service bot Loading…
@ZixuanJiang
Add numerics check for unreduced e2e.
#571 opened Jun 6, 2025 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.