Allows linalg structured operations to be handled during spmdization and sharding propagation. There is only support for projected permutation indexing maps.
1.1 KiB
1.1 KiB
Allows linalg structured operations to be handled during spmdization and sharding propagation. There is only support for projected permutation indexing maps.