[all-commits] [llvm/llvm-project] adbf21: [mlir][mesh] Add spmdization pass (#80518)

[Boian Petkantchin via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: adbf21f12b3069b2554efb39f2e92c6cf6f24940
      https://github.com/llvm/llvm-project/commit/adbf21f12b3069b2554efb39f2e92c6cf6f24940
  Author: Boian Petkantchin <boian.petkantchin at amd.com>
  Date:   2024-02-06 (Tue, 06 Feb 2024)

  Changed paths:
    A mlir/include/mlir/Dialect/Func/Extensions/MeshShardingExtensions.h
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h
    M mlir/include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.h
    M mlir/include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.td
    A mlir/include/mlir/Dialect/Mesh/Interfaces/ShardingInterfaceImpl.h
    M mlir/include/mlir/Dialect/Mesh/Transforms/Passes.td
    M mlir/include/mlir/Dialect/Mesh/Transforms/Spmdization.h
    M mlir/lib/Dialect/Func/Extensions/AllExtensions.cpp
    M mlir/lib/Dialect/Func/Extensions/CMakeLists.txt
    A mlir/lib/Dialect/Func/Extensions/MeshShardingExtensions.cpp
    M mlir/lib/Dialect/Mesh/IR/MeshOps.cpp
    M mlir/lib/Dialect/Mesh/Interfaces/ShardingInterface.cpp
    M mlir/lib/Dialect/Mesh/Transforms/ShardingPropagation.cpp
    M mlir/lib/Dialect/Mesh/Transforms/Spmdization.cpp
    M mlir/lib/Dialect/Tosa/IR/ShardingInterfaceImpl.cpp
    M mlir/test/Dialect/Mesh/resharding-spmdization.mlir
    A mlir/test/Dialect/Mesh/spmdization.mlir
    M utils/bazel/llvm-project-overlay/mlir/BUILD.bazel

  Log Message:
  -----------
  [mlir][mesh] Add spmdization pass (#80518)

Add a pass that converts a function that has sharding annotations into
SPMD form.