Pull requests: openxla/shardy
Pull requests list
#sdy add py binds for TensorShardingAttr::get with inlined mesh.
#122
opened Oct 4, 2024 by
copybara-service
bot
Loading…
[shardy] Fix a bug for non-minor-most factors in
BasicFactorPropagation::compatiblePrefixNoConflictsWithinFactor.
#121
opened Oct 4, 2024 by
copybara-service
bot
Loading…
#sdy Add support for specified an inlined
MeshAttr in a TensorShardingAttr instead of referencing a symbol MeshOp by name.
#120
opened Oct 3, 2024 by
copybara-service
bot
Loading…
[SDY] Add sharding group constraint enforcement to propagation.
#118
opened Oct 3, 2024 by
copybara-service
bot
Loading…
#sdy export NamedComputationOps with backend_configs to CallOps.
#117
opened Oct 1, 2024 by
copybara-service
bot
Loading…
#sdy import
CallOps with backend_configs to NamedComputationOps.
#116
opened Oct 1, 2024 by
copybara-service
bot
Loading…
Preserve HLO shardings on calls and non-entry functions.
#113
opened Sep 30, 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!
Exclude everything labeled
bug with -label:bug.