[all-commits] [llvm/llvm-project] 79aa77: [mlir][mesh] Add lowering of process multi-index o...

[Boian Petkantchin via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 79aa77626770c91badd7c9ba9d26e55a28d34416
      https://github.com/llvm/llvm-project/commit/79aa77626770c91badd7c9ba9d26e55a28d34416
  Author: Boian Petkantchin <boian.petkantchin at amd.com>
  Date:   2024-01-10 (Wed, 10 Jan 2024)

  Changed paths:
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.td
    A mlir/include/mlir/Dialect/Mesh/Transforms/Transforms.h
    M mlir/lib/Dialect/Mesh/IR/MeshOps.cpp
    M mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt
    M mlir/lib/Dialect/Mesh/Transforms/Simplifications.cpp
    M mlir/lib/Dialect/Mesh/Transforms/Spmdization.cpp
    A mlir/lib/Dialect/Mesh/Transforms/Transforms.cpp
    M mlir/test/Dialect/Mesh/invalid.mlir
    M mlir/test/Dialect/Mesh/ops.mlir
    A mlir/test/Dialect/Mesh/process-multi-index-op-lowering.mlir
    M mlir/test/Dialect/Mesh/resharding-spmdization.mlir
    M mlir/test/lib/Dialect/Mesh/CMakeLists.txt
    A mlir/test/lib/Dialect/Mesh/TestProcessMultiIndexOpLowering.cpp
    M mlir/tools/mlir-opt/mlir-opt.cpp

  Log Message:
  -----------
  [mlir][mesh] Add lowering of process multi-index op (#77490)

* Rename mesh.process_index -> mesh.process_multi_index.
* Add mesh.process_linear_index op.
* Add lowering of mesh.process_multi_index into an expression using
mesh.process_linear_index, mesh.cluster_shape and
affine.delinearize_index.

This is useful to lower mesh ops and prepare them for further lowering
where the runtime may have only the linear index of a device/process.
For example in MPI we have a rank (linear index) in a communicator.