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/ConvertToLLVM/ToLLVMInterface.h" 10 #include "mlir/Dialect/Arith/IR/Arith.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 21 using namespace mlir; 22 using namespace mlir::memref; 23 24 #include "mlir/Dialect/MemRef/IR/MemRefOpsDialect.cpp.inc" 25 26 //===----------------------------------------------------------------------===// 27 // MemRefDialect Dialect Interfaces 28 //===----------------------------------------------------------------------===// 29 30 namespace { 31 struct 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 44 void 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<MemRefDialect, ConvertToLLVMPatternInterface>(); 51 declarePromisedInterfaces<bufferization::AllocationOpInterface, AllocOp, 52 AllocaOp, ReallocOp>(); 53 declarePromisedInterfaces<RuntimeVerifiableOpInterface, CastOp, ExpandShapeOp, 54 LoadOp, ReinterpretCastOp, StoreOp, SubViewOp>(); 55 declarePromisedInterfaces<ValueBoundsOpInterface, AllocOp, AllocaOp, CastOp, 56 DimOp, GetGlobalOp, RankOp, SubViewOp>(); 57 declarePromisedInterface<MemRefType, DestructurableTypeInterface>(); 58 } 59 60 /// Finds the unique dealloc operation (if one exists) for `allocValue`. 61 std::optional<Operation *> mlir::memref::findDealloc(Value allocValue) { 62 Operation *dealloc = nullptr; 63 for (Operation *user : allocValue.getUsers()) { 64 if (!hasEffect<MemoryEffects::Free>(user, allocValue)) 65 continue; 66 // If we found a realloc instead of dealloc, return std::nullopt. 67 if (isa<memref::ReallocOp>(user)) 68 return std::nullopt; 69 // If we found > 1 dealloc, return std::nullopt. 70 if (dealloc) 71 return std::nullopt; 72 dealloc = user; 73 } 74 return dealloc; 75 } 76