522c1d0eea
This is necessary to support deallocation of IR with gpu.launch operations because it does not implement the RegionBranchOpInterface. Implementing the interface would require it to support regions with unstructured control flow and produced arguments/results.
38 lines
1.4 KiB
C++
38 lines
1.4 KiB
C++
//===- BufferDeallocationOpInterfaceImpl.cpp ------------------------------===//
|
|
//
|
|
// 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/GPU/Transforms/BufferDeallocationOpInterfaceImpl.h"
|
|
#include "mlir/Dialect/Bufferization/IR/BufferDeallocationOpInterface.h"
|
|
#include "mlir/Dialect/Bufferization/IR/Bufferization.h"
|
|
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::bufferization;
|
|
|
|
namespace {
|
|
///
|
|
struct GPUTerminatorOpInterface
|
|
: public BufferDeallocationOpInterface::ExternalModel<
|
|
GPUTerminatorOpInterface, gpu::TerminatorOp> {
|
|
FailureOr<Operation *> process(Operation *op, DeallocationState &state,
|
|
const DeallocationOptions &options) const {
|
|
SmallVector<Value> updatedOperandOwnerships;
|
|
return deallocation_impl::insertDeallocOpForReturnLike(
|
|
state, op, {}, updatedOperandOwnerships);
|
|
}
|
|
};
|
|
|
|
} // namespace
|
|
|
|
void mlir::gpu::registerBufferDeallocationOpInterfaceExternalModels(
|
|
DialectRegistry ®istry) {
|
|
registry.addExtension(+[](MLIRContext *ctx, GPUDialect *dialect) {
|
|
gpu::TerminatorOp::attachInterface<GPUTerminatorOpInterface>(*ctx);
|
|
});
|
|
}
|