08b7a71bcc
Since ownership based buffer deallocation requires a few passes to be run in a somewhat fixed sequence, it makes sense to have a pipeline for convenience (and to reduce the number of transform ops to represent default deallocation).
47 lines
2.1 KiB
C++
47 lines
2.1 KiB
C++
//===- BufferizationPipelines.cpp - Pipelines for bufferization -----------===//
|
|
//
|
|
// 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/Pipelines/Passes.h"
|
|
|
|
#include "mlir/Dialect/Bufferization/Transforms/Passes.h"
|
|
#include "mlir/Dialect/Func/IR/FuncOps.h"
|
|
#include "mlir/Dialect/MemRef/Transforms/Passes.h"
|
|
#include "mlir/Pass/PassManager.h"
|
|
#include "mlir/Transforms/Passes.h"
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Pipeline implementation.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
void mlir::bufferization::buildBufferDeallocationPipeline(
|
|
OpPassManager &pm, const BufferDeallocationPipelineOptions &options) {
|
|
pm.addNestedPass<func::FuncOp>(
|
|
memref::createExpandReallocPass(/*emitDeallocs=*/false));
|
|
pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
|
|
pm.addNestedPass<func::FuncOp>(createOwnershipBasedBufferDeallocationPass(
|
|
options.privateFunctionDynamicOwnership.getValue()));
|
|
pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
|
|
pm.addNestedPass<func::FuncOp>(createBufferDeallocationSimplificationPass());
|
|
pm.addPass(createLowerDeallocationsPass());
|
|
pm.addNestedPass<func::FuncOp>(createCSEPass());
|
|
pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Pipeline registration.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
void mlir::bufferization::registerBufferizationPipelines() {
|
|
PassPipelineRegistration<BufferDeallocationPipelineOptions>(
|
|
"buffer-deallocation-pipeline",
|
|
"The default pipeline for automatically inserting deallocation "
|
|
"operations after one-shot bufferization. Deallocation operations "
|
|
"(except `memref.realloc`) may not be present already.",
|
|
buildBufferDeallocationPipeline);
|
|
}
|