[all-commits] [llvm/llvm-project] 66c2b7: [MLIR] Extend vector.gather to accept tensor as base

Jerry Wu via All-commits all-commits at lists.llvm.org
Tue Aug 9 11:21:19 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 66c2b76846246fcd1df31e1928b0df44a8fe43e1
      https://github.com/llvm/llvm-project/commit/66c2b76846246fcd1df31e1928b0df44a8fe43e1
  Author: Jerry Wu <cheyuw at google.com>
  Date:   2022-08-09 (Tue, 09 Aug 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Vector/IR/VectorOps.td
    M mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp
    M mlir/lib/Dialect/Vector/IR/VectorOps.cpp
    M mlir/lib/Dialect/Vector/Transforms/BufferizableOpInterfaceImpl.cpp
    M mlir/test/Dialect/Vector/bufferize.mlir
    M mlir/test/Dialect/Vector/ops.mlir

  Log Message:
  -----------
  [MLIR] Extend vector.gather to accept tensor as base

In addition to memref, accept ranked tensor as the base operand of vector.gather, similar to vector.trasnfer_read.

This will allow us to vectorize noncontiguous tensor.extract into vector.gather. Full discussion can be found here: https://github.com/iree-org/iree/issues/9198

Reviewed By: hanchung, dcaballe

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




More information about the All-commits mailing list