[all-commits] [llvm/llvm-project] d4d289: [mlir][vector] Add distribution pattern for vector...

[Quinn Dawkins via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: d4d289144764f4d293874d8959cc0d65ff79148f
      https://github.com/llvm/llvm-project/commit/d4d289144764f4d293874d8959cc0d65ff79148f
  Author: Quinn Dawkins <quinn.dawkins at gmail.com>
  Date:   2023-11-10 (Fri, 10 Nov 2023)

  Changed paths:
    M mlir/lib/Dialect/Vector/Transforms/VectorDistribute.cpp
    M mlir/test/Dialect/Vector/vector-warp-distribute.mlir

  Log Message:
  -----------
  [mlir][vector] Add distribution pattern for vector.create_mask (#71619)

This is the last step needed for basic support for distributing masked
vector code. The lane id gets delinearized based on the distributed mask
shape and then compared against the original mask sizes to compute the
bounds for the distributed mask. Note that the distribution of masks is
implicit on the shape specified by the warp op. As a result, it is the
responsibility of the consumer of the mask to ensure the distributed
mask will match its own distribution semantics.