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