[all-commits] [llvm/llvm-project] 0fd50e: [MLIR][mesh] Mesh fixes (#124724)

[Frank Schlimbach via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0fd50ec9a3d6669b8fb835d125a0494ebeb558f0
      https://github.com/llvm/llvm-project/commit/0fd50ec9a3d6669b8fb835d125a0494ebeb558f0
  Author: Frank Schlimbach <frank.schlimbach at intel.com>
  Date:   2025-02-12 (Wed, 12 Feb 2025)

  Changed paths:
    A mlir/include/mlir/Dialect/Arith/Transforms/ShardingInterfaceImpl.h
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.td
    M mlir/include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.h
    M mlir/include/mlir/InitAllDialects.h
    M mlir/lib/Dialect/Arith/Transforms/CMakeLists.txt
    A mlir/lib/Dialect/Arith/Transforms/ShardingInterfaceImpl.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/Tensor/Extensions/MeshShardingExtensions.cpp
    A mlir/test/Dialect/Arith/mesh-spmdize.mlir
    A mlir/test/Dialect/Arith/sharding-propagation.mlir
    M mlir/test/Dialect/Mesh/canonicalization.mlir
    M mlir/test/Dialect/Mesh/ops.mlir
    M mlir/test/Dialect/Mesh/spmdization.mlir

  Log Message:
  -----------
  [MLIR][mesh] Mesh fixes (#124724)

A collection of fixes to the mesh dialect
- allow constants in sharding propagation/spmdization
- fixes to tensor replication (e.g. 0d tensors)
- improved canonicalization
- sharding propagation incorrectly generated too many ShardOps
New operation `mesh.GetShardOp` enables exchanging sharding information
(like on function boundaries)



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