Chengji Yao b0d5b4d252 [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.

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

3.9 KiB

Raw Blame History

View Raw