-
Notifications
You must be signed in to change notification settings - Fork 26
Pull requests: openxla/shardy
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Demote communication cost and only perform on global tensor sizes.
#908
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Do not remove RS for the pattern
AG-dot-RS in RemoveAllGatherReduceScatterForCMV1Pass.
#906
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Do not propagate shardings along reversed dimensions in Shardy.
#905
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Add support for custom sharding rules via ShardingRuleOpInterface
#885
opened Nov 14, 2025 by
sshonTT
Loading…
[mpmd] Move pipeline scheduling types and conversion functions to OSS
#884
opened Nov 14, 2025 by
copybara-service
bot
Loading…
[mpmd] Modify tests to also test custom pipeline API path
#883
opened Nov 14, 2025 by
copybara-service
bot
Loading…
[mpmd] Add custom pipeline API for user-defined scheduling and merging
#882
opened Nov 14, 2025 by
copybara-service
bot
Loading…
Add sharding rule for stablehlo.batch_norm_inference op
#847
opened Oct 15, 2025 by
sshonTT
Loading…
Update schedule in
op_priority_propagation.cc.
#814
opened Oct 4, 2025 by
copybara-service
bot
Loading…
[mpmd] Integrate PipelineSchedule API into jit lowering flow
#798
opened Sep 30, 2025 by
copybara-service
bot
Loading…
[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…
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…
Verify that the shrdings inside the
sdy.manual_computation shares the same mesh.
#624
opened Jul 23, 2025 by
copybara-service
bot
Loading…
#sdy Mark unreduced ops with frontend attributes.
#567
opened Jun 5, 2025 by
copybara-service
bot
Loading…
#sdy Add all-to-all fusion canonicalization pattern.
#516
opened May 7, 2025 by
copybara-service
bot
Loading…
Reverts 417dcc0d08153de3d533390db5258035e5869b54
#510
opened May 7, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Follow long discussions with comments:>50.