-
Notifications
You must be signed in to change notification settings - Fork 4
Pull requests: openxla/shardy
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[SDY] preprocess sharding groups which have shardings prior to propagation to validate there are no inter-group conflicts.
#107
opened Sep 19, 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] Refactor
propagateTensorSharding
methods to an external interface.
#82
opened Aug 28, 2024 by
copybara-service
bot
Loading…
#sdy Add custom-call sharding rule registry.
#50
opened Aug 8, 2024 by
copybara-service
bot
Loading…
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.