[all-commits] [llvm/llvm-project] 6e5732: [mlir][mesh, mpi] More on MeshToMPI (#129048)

[Frank Schlimbach via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 6e57326c61e77fe257d8a16adf9389231ea7c6ac
      https://github.com/llvm/llvm-project/commit/6e57326c61e77fe257d8a16adf9389231ea7c6ac
  Author: Frank Schlimbach <frank.schlimbach at intel.com>
  Date:   2025-03-06 (Thu, 06 Mar 2025)

  Changed paths:
    M mlir/include/mlir/Conversion/Passes.td
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.td
    M mlir/lib/Conversion/MeshToMPI/CMakeLists.txt
    M mlir/lib/Conversion/MeshToMPI/MeshToMPI.cpp
    M mlir/lib/Dialect/Mesh/IR/MeshOps.cpp
    M mlir/lib/Dialect/Tensor/Extensions/MeshShardingExtensions.cpp
    M mlir/test/Conversion/MeshToMPI/convert-mesh-to-mpi.mlir
    A mlir/test/Conversion/MeshToMPI/convert-shardshape-to-mpi.mlir
    M mlir/test/Dialect/Mesh/ops.mlir
    M mlir/test/Dialect/Tensor/mesh-spmdization.mlir

  Log Message:
  -----------
  [mlir][mesh, mpi] More on MeshToMPI (#129048)

- do not create MPI operations if no halo exchange is needed
- allow returning sharding information through `!mesh.sharding`
  (gets converted into a tuple of tensors)
- lowering `mesh.shard_shape` including fixes to the operation itself
- global symbol `static_mpi_rank` replaced by an DLTI attribute
  (now aligned with MPIToLLVM)
- smaller fixes and some minor cleanup

---------

Co-authored-by: Christian Ulmann <christianulmann at gmail.com>



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