[mlir][tensor] Fix verifier and bufferization of collapse_shape
authorMatthias Springer <springerm@google.com>
Fri, 8 Apr 2022 09:04:00 +0000 (18:04 +0900)
committerMatthias Springer <springerm@google.com>
Fri, 8 Apr 2022 09:20:40 +0000 (18:20 +0900)
commitd7a9bf91431a08bf43cc5b7111a043de9defaee9
tree1837c2977e38819b0878df18a0da3a363dd3c23a
parentd2608adf490c10afc71d57141d61a9df5464fd82
[mlir][tensor] Fix verifier and bufferization of collapse_shape

Insert a buffer copy unless the dims are guaranteed to be collapsible. In the verifier, accept collapses unless they are guaranteed to be non-collapsible.

Differential Revision: https://reviews.llvm.org/D123316
mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h
mlir/include/mlir/Dialect/MemRef/IR/MemRefOps.td
mlir/lib/Dialect/Bufferization/IR/BufferizableOpInterface.cpp
mlir/lib/Dialect/Linalg/Transforms/BufferizableOpInterfaceImpl.cpp
mlir/lib/Dialect/MemRef/IR/MemRefOps.cpp
mlir/lib/Dialect/Tensor/Transforms/BufferizableOpInterfaceImpl.cpp
mlir/test/Dialect/Tensor/bufferize.mlir