[all-commits] [llvm/llvm-project] d02f10: [mlir][vector] Add lowering pattern for vector.war...

[Thomas via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: d02f10d96d4506cfe2777e410570a47d83de214e
      https://github.com/llvm/llvm-project/commit/d02f10d96d4506cfe2777e410570a47d83de214e
  Author: Thomas Raoux <thomasraoux at google.com>
  Date:   2022-05-12 (Thu, 12 May 2022)

  Changed paths:
    A mlir/include/mlir/Dialect/Vector/Transforms/VectorDistribution.h
    M mlir/lib/Dialect/Vector/Transforms/CMakeLists.txt
    A mlir/lib/Dialect/Vector/Transforms/VectorDistribute.cpp
    A mlir/test/Dialect/Vector/vector-warp-distribute.mlir
    A mlir/test/Integration/Dialect/Vector/GPU/CUDA/lit.local.cfg
    A mlir/test/Integration/Dialect/Vector/GPU/CUDA/test-warp-distribute.mlir
    M mlir/test/lib/Dialect/Vector/CMakeLists.txt
    M mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
    M utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel

  Log Message:
  -----------
  [mlir][vector] Add lowering pattern for vector.warp_execute_on_lane_0 op

Add lowering of the vector.warp_execute_on_lane_0 into scf.if plus memory
transfer for the operands and yield values.

This also add an integration test running on GPU warp. The same tests can be
later re-used with different comment lines to tests distribution
transformations.

This is mostly from @springerm contribution.

Differential Revision: https://reviews.llvm.org/D125430