-
Notifications
You must be signed in to change notification settings - Fork 13
Pull requests: openxla/shardy
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
#sdy avoid sideways operand propagation for elementwise ops which can't propagate to the result.
#382
opened Feb 14, 2025 by
copybara-service
bot
Loading…
#sdy Add JAX backwards compatibility test.
#379
opened Feb 12, 2025 by
copybara-service
bot
Loading…
Replace std::bind with std::bind_front for PropagationDirectionAlongFactor.
#363
opened Feb 5, 2025 by
copybara-service
bot
Loading…
#sdy add sharding rules for ragged_all_to_all custom call
#338
opened Jan 23, 2025 by
copybara-service
bot
Loading…
Update to match upstream API change (NFC).
#282
opened Dec 23, 2024 by
copybara-service
bot
Loading…
#sdy fix constant splitter taking too long.
#251
opened Dec 6, 2024 by
copybara-service
bot
Loading…
Handle when JAX saves the manual axes on the
CallOp
of the shmap_body
.
#92
opened Sep 9, 2024 by
copybara-service
bot
Loading…
Support JAX shard map lowering with Shardy enabled.
#91
opened Sep 6, 2024 by
copybara-service
bot
Loading…
#sdy Add custom-call sharding rule registry.
#50
opened Aug 8, 2024 by
copybara-service
bot
Loading…
ProTip!
Add no:assignee to see everything that’s not assigned.