Pull requests: openxla/shardy
Pull requests list
#sdy #debug use
ShardingProjection so that we are able to correctly save debug info when there are sharding conflicts.
#293
opened Jan 8, 2025 by
copybara-service
bot
Loading…
Shardy. Mark the concat dimension as need replication.
#292
opened Jan 7, 2025 by
copybara-service
bot
Loading…
Update to match upstream API change (NFC).
#282
opened Dec 23, 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] 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!
Find all pull requests that aren't related to any open issues with -linked:issue.