[all-commits] [llvm/llvm-project] 73ddc4: [mlir][vector] Enable distribution over multiple d...

[Lei Zhang via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 73ddc4474bc4c34d3b6f50cd7a6e88a12ca83f8d
      https://github.com/llvm/llvm-project/commit/73ddc4474bc4c34d3b6f50cd7a6e88a12ca83f8d
  Author: Lei Zhang <antiagainst at google.com>
  Date:   2023-08-16 (Wed, 16 Aug 2023)

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

  Log Message:
  -----------
  [mlir][vector] Enable distribution over multiple dimensions

This commit starts enabling vector distruction over multiple
dimensions. It requires delinearize the lane ID to match the
expected rank. shape_cast and transfer_read now can properly
handle multiple dimensions.

Reviewed By: hanchung

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