[all-commits] [llvm/llvm-project] b0d5b4: [MLIR][Mesh] Add sharding propagation pass (#71261)

[Chengji Yao via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: b0d5b4d252357efbe5582ca635dd22c433ec02dc
      https://github.com/llvm/llvm-project/commit/b0d5b4d252357efbe5582ca635dd22c433ec02dc
  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 (#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.