[all-commits] [llvm/llvm-project] 1a8fb8: [mlir][mesh] Add resharding spmdization on a 1D de...

[Boian Petkantchin via All-commits]

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

  Changed paths:
    M mlir/include/mlir/Dialect/Mesh/IR/MeshBase.td
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.td
    A mlir/include/mlir/Dialect/Mesh/Transforms/ReshardingSpmdizationDoc.md
    A mlir/include/mlir/Dialect/Mesh/Transforms/Spmdization.h
    M mlir/lib/Dialect/Mesh/IR/MeshOps.cpp
    M mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt
    A mlir/lib/Dialect/Mesh/Transforms/Spmdization.cpp
    M mlir/test/Dialect/Mesh/invalid.mlir
    M mlir/test/Dialect/Mesh/ops.mlir
    A mlir/test/Dialect/Mesh/resharding-spmdization.mlir
    M mlir/test/lib/Dialect/Mesh/CMakeLists.txt
    A mlir/test/lib/Dialect/Mesh/TestReshardingSpmdization.cpp
    M mlir/tools/mlir-opt/mlir-opt.cpp

  Log Message:
  -----------
  [mlir][mesh] Add resharding spmdization on a 1D device mesh (#76179)

The current implementation supports only sharding of tensor axes that
have size divisible by the mesh axis size.