[all-commits] [llvm/llvm-project] 9d9400: [MLIR][Mesh] Add sharding propagation pass (#69665)

[Chengji Yao via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9d9400d7de9b928e3018af97e8b381a4a6ba5162
      https://github.com/llvm/llvm-project/commit/9d9400d7de9b928e3018af97e8b381a4a6ba5162
  Author: Chengji Yao <yaochengji at hotmail.com>
  Date:   2023-11-03 (Fri, 03 Nov 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Mesh/CMakeLists.txt
    M mlir/include/mlir/Dialect/Mesh/IR/MeshBase.td
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h
    A mlir/include/mlir/Dialect/Mesh/Interfaces/CMakeLists.txt
    A mlir/include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.h
    A mlir/include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.td
    A mlir/include/mlir/Dialect/Mesh/Transforms/CMakeLists.txt
    A mlir/include/mlir/Dialect/Mesh/Transforms/Passes.h
    A mlir/include/mlir/Dialect/Mesh/Transforms/Passes.td
    A mlir/include/mlir/Dialect/Tosa/IR/ShardingInterfaceImpl.h
    M mlir/include/mlir/IR/AffineMap.h
    M mlir/include/mlir/InitAllDialects.h
    M mlir/include/mlir/InitAllPasses.h
    M mlir/lib/Dialect/Mesh/CMakeLists.txt
    M mlir/lib/Dialect/Mesh/IR/MeshOps.cpp
    A mlir/lib/Dialect/Mesh/Interfaces/CMakeLists.txt
    A mlir/lib/Dialect/Mesh/Interfaces/ShardingInterface.cpp
    A mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt
    A mlir/lib/Dialect/Mesh/Transforms/ShardingPropagation.cpp
    M mlir/lib/Dialect/Tosa/CMakeLists.txt
    A mlir/lib/Dialect/Tosa/IR/ShardingInterfaceImpl.cpp
    M mlir/lib/IR/AffineMap.cpp
    A mlir/test/Dialect/Mesh/sharding-propagation.mlir

  Log Message:
  -----------
  [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.