A machine learning compiler for GPUs, CPUs, and ML accelerators
2.39k
stars
356
forks
source link
[XLA:SPMD] Fix sharding propagation for kGetTupleElement not generating correct sharding shape with subgroup manual. #14304
Open
copybara-service[bot] opened 2 days ago
[XLA:SPMD] Fix sharding propagation for kGetTupleElement not generating correct sharding shape with subgroup manual.