-
Notifications
You must be signed in to change notification settings - Fork 7
Pull requests: openxla/shardy
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add an option to skip converting sharding_constraint to reshard
#220
opened Nov 18, 2024 by
copybara-service
bot
Loading…
[SDY] refactor propagation functions in basic_propagation.cc to utilize a parameter struct (to cleanup function signatures).
#192
opened Nov 10, 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!
no:milestone will show everything without a milestone.