A machine learning compiler for GPUs, CPUs, and ML accelerators
2.55k
stars
394
forks
source link
[XLA:SPMD] Allow common sharding candidates from shard_as group to be empty as sharding-aware CSE in sharding propagation. #14882
Closed
copybara-service[bot] closed 1 month ago
[XLA:SPMD] Allow common sharding candidates from shard_as group to be empty as sharding-aware CSE in sharding propagation.