[all-commits] [llvm/llvm-project] fb582b: [mlir] Implement Mesh's ShardingInterface for Lina...

Boian Petkantchin via All-commits all-commits at lists.llvm.org
Thu Mar 7 17:06:06 PST 2024


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: fb582b6ace781ff6991775d6dcd4df98aa16698f
      https://github.com/llvm/llvm-project/commit/fb582b6ace781ff6991775d6dcd4df98aa16698f
  Author: Boian Petkantchin <boian.petkantchin at amd.com>
  Date:   2024-03-07 (Thu, 07 Mar 2024)

  Changed paths:
    A mlir/include/mlir/Dialect/Linalg/Transforms/AllInterfaces.h
    A mlir/include/mlir/Dialect/Linalg/Transforms/MeshShardingInterfaceImpl.h
    M mlir/include/mlir/Dialect/Mesh/IR/MeshBase.td
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.td
    M mlir/include/mlir/Dialect/Mesh/Interfaces/ShardingInterfaceImpl.h
    M mlir/include/mlir/Dialect/Mesh/Transforms/Transforms.h
    M mlir/include/mlir/IR/Dialect.h
    M mlir/include/mlir/InitAllDialects.h
    M mlir/lib/Dialect/Linalg/IR/CMakeLists.txt
    M mlir/lib/Dialect/Linalg/IR/LinalgDialect.cpp
    A mlir/lib/Dialect/Linalg/Transforms/AllInterfaces.cpp
    M mlir/lib/Dialect/Linalg/Transforms/CMakeLists.txt
    A mlir/lib/Dialect/Linalg/Transforms/MeshShardingInterfaceImpl.cpp
    M mlir/lib/Dialect/Linalg/Transforms/TilingInterfaceImpl.cpp
    M mlir/lib/Dialect/Mesh/IR/MeshOps.cpp
    M mlir/lib/Dialect/Mesh/Interfaces/ShardingInterface.cpp
    M mlir/lib/Dialect/Mesh/Transforms/Transforms.cpp
    A mlir/test/Dialect/Linalg/mesh-spmdization.mlir
    M utils/bazel/llvm-project-overlay/mlir/BUILD.bazel

  Log Message:
  -----------
  [mlir] Implement Mesh's ShardingInterface for Linalg ops (#82284)

Allows linalg structured operations to be handled during spmdization and
sharding propagation.

There is only support for projected permutation indexing maps.



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications


More information about the All-commits mailing list