//===- TestTensorCopyInsertion.cpp - Bufferization Analysis -----*- c++ -*-===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // //===----------------------------------------------------------------------===// #include "mlir/Dialect/Bufferization/IR/Bufferization.h" #include "mlir/Dialect/Bufferization/Transforms/OneShotAnalysis.h" #include "mlir/Dialect/Bufferization/Transforms/Transforms.h" #include "mlir/Pass/Pass.h" usingnamespacemlir; namespace { /// This pass runs One-Shot Analysis and inserts copies for all OpOperands that /// were decided to bufferize out-of-place. After running this pass, a /// bufferization can write to buffers directly (without making copies) and no /// longer has to care about potential read-after-write conflicts. /// /// Note: By default, all newly inserted tensor copies/allocs (i.e., newly /// created `bufferization.alloc_tensor` ops) that do not escape block are /// annotated with `escape = false`. If `create-allocs` is unset, all newly /// inserted tensor copies/allocs are annotated with `escape = true`. In that /// case, they are not getting deallocated when bufferizing the IR. struct TestTensorCopyInsertionPass : public PassWrapper<TestTensorCopyInsertionPass, OperationPass<ModuleOp>> { … }; } // namespace namespace mlir::test { void registerTestTensorCopyInsertionPass() { … } } // namespace mlir::test