[mlir][SCF] Fix scf.while bufferization
authorMatthias Springer <springerm@google.com>
Tue, 17 May 2022 20:58:54 +0000 (22:58 +0200)
committerMatthias Springer <springerm@google.com>
Tue, 17 May 2022 22:35:50 +0000 (00:35 +0200)
commit996834e6813ab5481a58e42e7a11f57d243a3a99
treef912b6dd3e12720510d863c49db4e92de23d227b
parent79ca4ed3e782e505e8964a3c968de5fd4f09ca7a
[mlir][SCF] Fix scf.while bufferization

Before this fix, the bufferization implementation made the incorrect assumption that the values yielded from the "before" region must match with the values yielded from the "after" region.

Differential Revision: https://reviews.llvm.org/D125835
mlir/include/mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h
mlir/lib/Dialect/Bufferization/IR/BufferizableOpInterface.cpp
mlir/lib/Dialect/SCF/Transforms/BufferizableOpInterfaceImpl.cpp
mlir/lib/Dialect/Tensor/Transforms/BufferizableOpInterfaceImpl.cpp
mlir/test/Dialect/SCF/one-shot-bufferize.mlir