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

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

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.

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 87f5e229
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