[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