Unverified Commit 9d9400d7 authored by Chengji Yao's avatar Chengji Yao Committed by GitHub
Browse files

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

Add a pass that propagates sharding information throughout the graph.
After this pass, each of the operations' operands and results is
annotated with a `mesh.shard` operation, and the operations themselves
are added with sharding option attributes.

The pass is driven by  a newly added `ShardingInterface`, and an implementation
for element-wise and matmul ops in the TOSA dialect is provided.
parent 946b33d3
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