1//===----------------------------------------------------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "mlir/Conversion/ConvertToEmitC/ToEmitCInterface.h"
10#include "mlir/Conversion/ConvertToLLVM/ToLLVMInterface.h"
11#include "mlir/Dialect/Bufferization/IR/AllocationOpInterface.h"
12#include "mlir/Dialect/MemRef/IR/MemRef.h"
13#include "mlir/IR/BuiltinTypes.h"
14#include "mlir/Interfaces/MemorySlotInterfaces.h"
15#include "mlir/Interfaces/RuntimeVerifiableOpInterface.h"
16#include "mlir/Interfaces/SideEffectInterfaces.h"
17#include "mlir/Interfaces/ValueBoundsOpInterface.h"
18#include "mlir/Transforms/InliningUtils.h"
19#include <optional>
20
21using namespace mlir;
22using namespace mlir::memref;
23
24#include "mlir/Dialect/MemRef/IR/MemRefOpsDialect.cpp.inc"
25
26//===----------------------------------------------------------------------===//
27// MemRefDialect Dialect Interfaces
28//===----------------------------------------------------------------------===//
29
30namespace {
31struct MemRefInlinerInterface : public DialectInlinerInterface {
32 using DialectInlinerInterface::DialectInlinerInterface;
33 bool isLegalToInline(Region *dest, Region *src, bool wouldBeCloned,
34 IRMapping &valueMapping) const final {
35 return true;
36 }
37 bool isLegalToInline(Operation *, Region *, bool wouldBeCloned,
38 IRMapping &) const final {
39 return true;
40 }
41};
42} // namespace
43
44void mlir::memref::MemRefDialect::initialize() {
45 addOperations<
46#define GET_OP_LIST
47#include "mlir/Dialect/MemRef/IR/MemRefOps.cpp.inc"
48 >();
49 addInterfaces<MemRefInlinerInterface>();
50 declarePromisedInterface<ConvertToEmitCPatternInterface, MemRefDialect>();
51 declarePromisedInterface<ConvertToLLVMPatternInterface, MemRefDialect>();
52 declarePromisedInterfaces<bufferization::AllocationOpInterface, AllocOp,
53 AllocaOp, ReallocOp>();
54 declarePromisedInterfaces<RuntimeVerifiableOpInterface, AssumeAlignmentOp,
55 AtomicRMWOp, CastOp, CopyOp, DimOp, ExpandShapeOp,
56 GenericAtomicRMWOp, LoadOp, StoreOp, SubViewOp>();
57 declarePromisedInterfaces<ValueBoundsOpInterface, AllocOp, AllocaOp, CastOp,
58 DimOp, GetGlobalOp, RankOp, SubViewOp>();
59 declarePromisedInterface<DestructurableTypeInterface, MemRefType>();
60}
61
62/// Finds the unique dealloc operation (if one exists) for `allocValue`.
63std::optional<Operation *> mlir::memref::findDealloc(Value allocValue) {
64 Operation *dealloc = nullptr;
65 for (Operation *user : allocValue.getUsers()) {
66 if (!hasEffect<MemoryEffects::Free>(op: user, value: allocValue))
67 continue;
68 // If we found a realloc instead of dealloc, return std::nullopt.
69 if (isa<memref::ReallocOp>(Val: user))
70 return std::nullopt;
71 // If we found > 1 dealloc, return std::nullopt.
72 if (dealloc)
73 return std::nullopt;
74 dealloc = user;
75 }
76 return dealloc;
77}
78

source code of mlir/lib/Dialect/MemRef/IR/MemRefDialect.cpp