Boian Petkantchin d635b860f3

[mlir][mesh] Insert resharding during sharding propagation (#84514) ...

If there are conflicts between the sharding annotations of some op,
insert resharding.
Make the Spmdization pass more forgiving to allow for more than 2
chained `mesh.shard` ops.

Implement `getReductionLoopIteratorKinds` in ShardingInterface for
linalg ops.

2024-05-22 13:33:25 -07:00

..

CMakeLists.txt

[MLIR][Mesh] Add sharding propagation pass (#71261)

2023-11-03 21:07:31 -07:00

ShardingInterface.cpp

[mlir][mesh] Insert resharding during sharding propagation (#84514)

2024-05-22 13:33:25 -07:00