Unverified Commit d635b860 authored by Boian Petkantchin's avatar Boian Petkantchin Committed by GitHub
Browse files

[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.
parent bd3f5a4b
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment