Pull requests: openxla/shardy
Pull requests list
#sdy. Update sharding rule builder and registry to support factor types.
#284
opened Dec 28, 2024 by
copybara-service
bot
Loading…
Update to match upstream API change (NFC).
#282
opened Dec 23, 2024 by
copybara-service
bot
Loading…
#sdy use
applyPatternsGreedily with config.fold=false and config.cseConstants=false to avoid constant folding and CSE which is expensive.
#281
opened Dec 21, 2024 by
copybara-service
bot
Loading…
#sdy support JAX callbacks through the Shardy XLA round-trip pipeline.
#272
opened Dec 16, 2024 by
copybara-service
bot
Loading…
#sdy fix constant splitter taking too long.
#251
opened Dec 6, 2024 by
copybara-service
bot
Loading…
#sdy #debug Support functions with multiple results and
ShardableDataFlowOpInterface ops.
#247
opened Dec 2, 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!
Add no:assignee to see everything that’s not assigned.