[all-commits] [llvm/llvm-project] 4b3446: [mlir][mesh] Add endomorphism simplification for a...

[Boian Petkantchin via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 4b3446771f745bb5169354ad9027c0a1c9fca394
      https://github.com/llvm/llvm-project/commit/4b3446771f745bb5169354ad9027c0a1c9fca394
  Author: Boian Petkantchin <boian.petkantchin at amd.com>
  Date:   2023-12-12 (Tue, 12 Dec 2023)

  Changed paths:
    A mlir/include/mlir/Dialect/Mesh/Transforms/Simplifications.h
    A mlir/include/mlir/Transforms/EndomorphismSimplification.h
    A mlir/include/mlir/Transforms/HomomorphismSimplification.h
    M mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt
    A mlir/lib/Dialect/Mesh/Transforms/Simplifications.cpp
    A mlir/test/Dialect/Mesh/simplifications.mlir
    M mlir/test/lib/Dialect/CMakeLists.txt
    A mlir/test/lib/Dialect/Mesh/CMakeLists.txt
    A mlir/test/lib/Dialect/Mesh/TestSimplifications.cpp
    M mlir/tools/mlir-opt/CMakeLists.txt
    M mlir/tools/mlir-opt/mlir-opt.cpp

  Log Message:
  -----------
  [mlir][mesh] Add endomorphism simplification for all-reduce (#73150)

Does transformations like
all_reduce(x) + all_reduce(y) -> all_reduce(x + y)

max(all_reduce(x), all_reduce(y)) -> all_reduce(max(x, y))
when the all_reduce element-wise op is max.

Added general rewrite pattern HomomorphismSimplification and
EndomorphismSimplification that encapsulate the general algorithm.
Made specialization for all-reduce with respect to
addf, addi, minsi, maxsi, minimumf and maximumf
in the Arithmetic dialect.