Lines Matching defs:moduleOp

49 mlir::LLVM::lookupOrCreateFn(Operation *moduleOp, StringRef name,
52   assert(moduleOp->hasTrait<OpTrait::SymbolTable>() &&
55       SymbolTable::lookupSymbolIn(moduleOp, name));
72   OpBuilder b(moduleOp->getRegion(0));
74       moduleOp->getLoc(), name,
79 lookupOrCreateReservedFn(Operation *moduleOp, StringRef name,
81   return lookupOrCreateFn(moduleOp, name, paramTypes, resultType,
86 mlir::LLVM::lookupOrCreatePrintI64Fn(Operation *moduleOp) {
88       moduleOp, kPrintI64, IntegerType::get(moduleOp->getContext(), 64),
89       LLVM::LLVMVoidType::get(moduleOp->getContext()));
93 mlir::LLVM::lookupOrCreatePrintU64Fn(Operation *moduleOp) {
95       moduleOp, kPrintU64, IntegerType::get(moduleOp->getContext(), 64),
96       LLVM::LLVMVoidType::get(moduleOp->getContext()));
100 mlir::LLVM::lookupOrCreatePrintF16Fn(Operation *moduleOp) {
102       moduleOp, kPrintF16,
103       IntegerType::get(moduleOp->getContext(), 16), // bits!
104       LLVM::LLVMVoidType::get(moduleOp->getContext()));
108 mlir::LLVM::lookupOrCreatePrintBF16Fn(Operation *moduleOp) {
110       moduleOp, kPrintBF16,
111       IntegerType::get(moduleOp->getContext(), 16), // bits!
112       LLVM::LLVMVoidType::get(moduleOp->getContext()));
116 mlir::LLVM::lookupOrCreatePrintF32Fn(Operation *moduleOp) {
118       moduleOp, kPrintF32, Float32Type::get(moduleOp->getContext()),
119       LLVM::LLVMVoidType::get(moduleOp->getContext()));
123 mlir::LLVM::lookupOrCreatePrintF64Fn(Operation *moduleOp) {
125       moduleOp, kPrintF64, Float64Type::get(moduleOp->getContext()),
126       LLVM::LLVMVoidType::get(moduleOp->getContext()));
139     Operation *moduleOp, std::optional<StringRef> runtimeFunctionName) {
141       moduleOp, runtimeFunctionName.value_or(kPrintString),
142       getCharPtr(moduleOp->getContext()),
143       LLVM::LLVMVoidType::get(moduleOp->getContext()));
147 mlir::LLVM::lookupOrCreatePrintOpenFn(Operation *moduleOp) {
149       moduleOp, kPrintOpen, {},
150       LLVM::LLVMVoidType::get(moduleOp->getContext()));
154 mlir::LLVM::lookupOrCreatePrintCloseFn(Operation *moduleOp) {
156       moduleOp, kPrintClose, {},
157       LLVM::LLVMVoidType::get(moduleOp->getContext()));
161 mlir::LLVM::lookupOrCreatePrintCommaFn(Operation *moduleOp) {
163       moduleOp, kPrintComma, {},
164       LLVM::LLVMVoidType::get(moduleOp->getContext()));
168 mlir::LLVM::lookupOrCreatePrintNewlineFn(Operation *moduleOp) {
170       moduleOp, kPrintNewline, {},
171       LLVM::LLVMVoidType::get(moduleOp->getContext()));
175 mlir::LLVM::lookupOrCreateMallocFn(Operation *moduleOp, Type indexType) {
176   return lookupOrCreateReservedFn(moduleOp, kMalloc, indexType,
177                                   getVoidPtr(moduleOp->getContext()));
181 mlir::LLVM::lookupOrCreateAlignedAllocFn(Operation *moduleOp, Type indexType) {
182   return lookupOrCreateReservedFn(moduleOp, kAlignedAlloc,
184                                   getVoidPtr(moduleOp->getContext()));
188 mlir::LLVM::lookupOrCreateFreeFn(Operation *moduleOp) {
190       moduleOp, kFree, getVoidPtr(moduleOp->getContext()),
191       LLVM::LLVMVoidType::get(moduleOp->getContext()));
195 mlir::LLVM::lookupOrCreateGenericAllocFn(Operation *moduleOp, Type indexType) {
196   return lookupOrCreateReservedFn(moduleOp, kGenericAlloc, indexType,
197                                   getVoidPtr(moduleOp->getContext()));
201 mlir::LLVM::lookupOrCreateGenericAlignedAllocFn(Operation *moduleOp,
203   return lookupOrCreateReservedFn(moduleOp, kGenericAlignedAlloc,
205                                   getVoidPtr(moduleOp->getContext()));
209 mlir::LLVM::lookupOrCreateGenericFreeFn(Operation *moduleOp) {
211       moduleOp, kGenericFree, getVoidPtr(moduleOp->getContext()),
212       LLVM::LLVMVoidType::get(moduleOp->getContext()));
216 mlir::LLVM::lookupOrCreateMemRefCopyFn(Operation *moduleOp, Type indexType,
219       moduleOp, kMemRefCopy,
221       LLVM::LLVMVoidType::get(moduleOp->getContext()));