Pull requests: openxla/shardy
Pull requests list
Move verification diagnostics in
shardy/dialect/sdy/ir/test/constant_verification.mlir.
#393
by copybara-service
bot
was merged Feb 24, 2025
Loading…
#sdy Only propagate backwards through
stablehlo.broadcast_in_dim in the 2nd op-priority iteration.
#392
by copybara-service
bot
was merged Feb 24, 2025
Loading…
Allow
sdy.collective_permute to reorder device ids in the mesh and add support in -sdy-reshard-to-collectives.
#391
by copybara-service
bot
was closed Feb 21, 2025
Loading…
#sdy when inserting all-to-alls, first try to insert all-to-alls that move axes to the right position at the target dimension.
#386
by copybara-service
bot
was closed Feb 17, 2025
Loading…
Tag the reverse dimensions as kPermutation, since the SPMD partitioner adds halo exchange if the reverse dimensions is sharded.
#385
by copybara-service
bot
was closed Feb 14, 2025
Loading…
#sdy Add an event to the op priority schedule to propagate through all ops after element-wise ops, but **only** through non-passthrough factors.
#384
by copybara-service
bot
was closed Feb 18, 2025
Loading…
#sdy avoid sideways operand propagation for elementwise ops which can't propagate to the result.
#382
by copybara-service
bot
was closed Feb 20, 2025
Loading…
Fix broken links in sharding_representation.md
#381
by copybara-service
bot
was merged Feb 14, 2025
Loading…
Add a
OpShardingRuleAttr::getFactorType and rename FactorType::kDefault to FactorType::kPassThrough
#380
by copybara-service
bot
was closed Feb 13, 2025
Loading…
Create variable from structured binding so that it can be captured in lambda.
#376
by copybara-service
bot
was merged Feb 10, 2025
Loading…
[Shardy] Replace "mhlo" with "stablehlo" at docs
#374
by copybara-service
bot
was closed Feb 10, 2025
Loading…
#sdy add reduction and permutation factor types to relevant factors
#373
by copybara-service
bot
was merged Feb 12, 2025
Loading…
#sdy Pass
getDirectionToPropagate into PropagateDataFlowEdgeOp such that we can control the direction for data flow edge op.
#368
by copybara-service
bot
was merged Feb 5, 2025
Loading…
ProTip!
Follow long discussions with comments:>50.