Home
last modified time | relevance | path

Searched refs:LLVMFuncOp (Results 1 – 25 of 28) sorted by relevance

12

/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DOpToFuncCallLowering.h43 using LLVM::LLVMFuncOp; in matchAndRewrite()
65 LLVMFuncOp funcOp = appendOrGetFuncOp(funcName, funcType, op); in matchAndRewrite()
111 LLVM::LLVMFuncOp appendOrGetFuncOp(StringRef funcName, in appendOrGetFuncOp()
114 using LLVM::LLVMFuncOp; in appendOrGetFuncOp()
118 return cast<LLVMFuncOp>(*funcOp); in appendOrGetFuncOp()
120 mlir::OpBuilder b(op->getParentOfType<LLVMFuncOp>()); in appendOrGetFuncOp()
121 return b.create<LLVMFuncOp>(op->getLoc(), funcName, funcType); in appendOrGetFuncOp()
DGPUOpsLowering.h77 auto llvmFuncOp = rewriter.create<LLVM::LLVMFuncOp>( in matchAndRewrite()
DConvertLaunchFuncToRuntimeCalls.cpp289 if (auto function = module.lookupSymbol<LLVM::LLVMFuncOp>(functionName)) in create()
292 .create<LLVM::LLVMFuncOp>(loc, functionName, functionType); in create()
/external/llvm-project/mlir/lib/Conversion/GPUToVulkan/
DConvertLaunchFuncToVulkanCalls.cpp290 builder.create<LLVM::LLVMFuncOp>( in declareVulkanFunctions()
298 builder.create<LLVM::LLVMFuncOp>( in declareVulkanFunctions()
307 builder.create<LLVM::LLVMFuncOp>( in declareVulkanFunctions()
315 builder.create<LLVM::LLVMFuncOp>( in declareVulkanFunctions()
337 builder.create<LLVM::LLVMFuncOp>(loc, fnName, fnType); in declareVulkanFunctions()
343 builder.create<LLVM::LLVMFuncOp>( in declareVulkanFunctions()
350 builder.create<LLVM::LLVMFuncOp>( in declareVulkanFunctions()
/external/llvm-project/mlir/lib/ExecutionEngine/
DJitRunner.cpp188 auto mainFunction = module.lookupSymbol<LLVM::LLVMFuncOp>(entryPoint); in compileAndExecuteVoidFunction()
196 Error checkCompatibleReturnType(LLVM::LLVMFuncOp mainFunction);
198 Error checkCompatibleReturnType<int32_t>(LLVM::LLVMFuncOp mainFunction) { in checkCompatibleReturnType()
204 Error checkCompatibleReturnType<int64_t>(LLVM::LLVMFuncOp mainFunction) { in checkCompatibleReturnType()
210 Error checkCompatibleReturnType<float>(LLVM::LLVMFuncOp mainFunction) { in checkCompatibleReturnType()
219 auto mainFunction = module.lookupSymbol<LLVM::LLVMFuncOp>(entryPoint); in compileAndExecuteSingleReturnFunction()
/external/llvm-project/mlir/lib/Target/LLVMIR/
DDebugTranslation.h26 class LLVMFuncOp; variable
40 void translate(LLVMFuncOp func, llvm::Function &llvmFunc);
DConvertToNVVMIR.cpp79 ModuleTranslation::getModuleBody(m).getOps<LLVM::LLVMFuncOp>()) { in translateModuleToNVVMIR()
DConvertToROCDLIR.cpp88 ModuleTranslation::getModuleBody(m).getOps<LLVM::LLVMFuncOp>()) { in translateModuleToROCDLIR()
DModuleTranslation.cpp713 LLVM::LLVMFuncOp function = addressOfOp.getFunction(); in convertOperation()
895 LogicalResult ModuleTranslation::convertOneFunction(LLVMFuncOp func) { in convertOneFunction()
973 if (!isa<LLVM::LLVMFuncOp, LLVM::GlobalOp>(&o) && !o.isKnownTerminator()) in checkSupportedModuleOps()
981 for (auto function : getModuleBody(mlirModule).getOps<LLVMFuncOp>()) { in convertFunctionSignatures()
1000 for (auto function : getModuleBody(mlirModule).getOps<LLVMFuncOp>()) { in convertFunctions()
DDebugTranslation.cpp79 void DebugTranslation::translate(LLVMFuncOp func, llvm::Function &llvmFunc) { in translate()
DConvertFromLLVMIR.cpp111 while (!isa<LLVMFuncOp, ModuleTerminatorOp>(i)) in getGlobalInsertPt()
785 LLVMFuncOp fop = in processFunction()
786 b.create<LLVMFuncOp>(UnknownLoc::get(context), f->getName(), functionType, in processFunction()
/external/llvm-project/mlir/include/mlir/Target/LLVMIR/
DModuleTranslation.h43 class LLVMFuncOp; variable
120 LogicalResult convertOneFunction(LLVMFuncOp func);
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_kernel_to_llvm_pass.cc195 launch_op->getParentOfType<LLVM::LLVMFuncOp>().getArgument(0); in matchAndRewrite()
198 auto function = SymbolTable::lookupNearestSymbolFrom<LLVM::LLVMFuncOp>( in matchAndRewrite()
218 function = rewriter.create<LLVM::LLVMFuncOp>( in matchAndRewrite()
Dtf_framework_legalize_to_llvm.cc33 using LLVM::LLVMFuncOp;
51 auto tf_func = module.lookupSymbol<LLVMFuncOp>(tf_func_name); in getOrInsertTFFunction()
56 tf_func = rewriter.create<LLVMFuncOp>(rewriter.getUnknownLoc(), in getOrInsertTFFunction()
Dtensorflow_abi_knowledge_propagation.cc85 auto kernel = module.lookupSymbol<LLVM::LLVMFuncOp>(launch.kernel()); in runOnFunction()
Dsame_shape_propagation.cc325 auto kernel = module.lookupSymbol<LLVM::LLVMFuncOp>(launch.kernel()); in runOnFunction()
/external/llvm-project/mlir/examples/toy/Ch7/mlir/
DLowerToLLVM.cpp109 if (module.lookupSymbol<LLVM::LLVMFuncOp>("printf")) in getOrInsertPrintf()
122 rewriter.create<LLVM::LLVMFuncOp>(module.getLoc(), "printf", llvmFnType); in getOrInsertPrintf()
/external/llvm-project/mlir/examples/toy/Ch6/mlir/
DLowerToLLVM.cpp109 if (module.lookupSymbol<LLVM::LLVMFuncOp>("printf")) in getOrInsertPrintf()
122 rewriter.create<LLVM::LLVMFuncOp>(module.getLoc(), "printf", llvmFnType); in getOrInsertPrintf()
/external/llvm-project/mlir/lib/Dialect/LLVMIR/Transforms/
DLegalizeForExport.cpp52 op->walk([](LLVMFuncOp f) { in ensureDistinctSuccessors()
/external/llvm-project/mlir/lib/Dialect/LLVMIR/IR/
DLLVMDialect.cpp460 if (LLVMFuncOp func = op->getParentOfType<LLVMFuncOp>()) { in verify()
580 auto fn = dyn_cast<LLVMFuncOp>(callee); in verify()
992 LLVMFuncOp AddressOfOp::getFunction() { in getFunction()
993 return lookupSymbolInModule<LLVM::LLVMFuncOp>((*this)->getParentOp(), in getFunction()
1290 Block *LLVMFuncOp::addEntryBlock() { in addEntryBlock()
1303 void LLVMFuncOp::build(OpBuilder &builder, OperationState &result, in build()
1414 static void printLLVMFuncOp(OpAsmPrinter &p, LLVMFuncOp op) { in printLLVMFuncOp()
1445 LogicalResult LLVMFuncOp::verifyType() { in verifyType()
1456 unsigned LLVMFuncOp::getNumFuncArguments() { in getNumFuncArguments()
1462 unsigned LLVMFuncOp::getNumFuncResults() { in getNumFuncResults()
[all …]
/external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/
DConvertLaunchFuncToLLVMCalls.cpp180 auto kernelFunc = module.lookupSymbol<LLVM::LLVMFuncOp>(newKernelFuncName); in matchAndRewrite()
184 kernelFunc = rewriter.create<LLVM::LLVMFuncOp>( in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/StandardToLLVM/
DStandardToLLVM.cpp1232 FuncOp funcOp, LLVM::LLVMFuncOp newFuncOp) { in wrapForExternalCallers()
1236 auto wrapperFuncOp = rewriter.create<LLVM::LLVMFuncOp>( in wrapForExternalCallers()
1273 FuncOp funcOp, LLVM::LLVMFuncOp newFuncOp) { in wrapExternalFunction()
1287 auto wrapperFunc = builder.create<LLVM::LLVMFuncOp>( in wrapExternalFunction()
1347 LLVM::LLVMFuncOp
1381 auto newFuncOp = rewriter.create<LLVM::LLVMFuncOp>( in convertFuncOpToLLVMFuncOp()
1721 auto abortFunc = module.lookupSymbol<LLVM::LLVMFuncOp>("abort"); in matchAndRewrite()
1727 abortFunc = rewriter.create<LLVM::LLVMFuncOp>(rewriter.getUnknownLoc(), in matchAndRewrite()
1946 auto allocFuncOp = module.lookupSymbol<LLVM::LLVMFuncOp>(name); in createAllocCall()
1955 allocFuncOp = rewriter.create<LLVM::LLVMFuncOp>(rewriter.getUnknownLoc(), in createAllocCall()
[all …]
/external/llvm-project/mlir/lib/Conversion/AsyncToLLVM/
DAsyncToLLVM.cpp158 builder.create<LLVM::LLVMFuncOp>(module.getLoc(), name, type); in addLLVMFuncDecl()
230 auto resumeOp = moduleBuilder.create<LLVM::LLVMFuncOp>( in addResumeFunction()
/external/llvm-project/mlir/docs/Tutorials/Toy/
DCh-6.md35 if (module.lookupSymbol<LLVM::LLVMFuncOp>("printf"))
48 rewriter.create<LLVM::LLVMFuncOp>(module.getLoc(), "printf", llvmFnType);
/external/llvm-project/mlir/include/mlir/Dialect/LLVMIR/
DLLVMOps.td413 OpBuilderDAG<(ins "LLVMFuncOp":$func, "ValueRange":$operands,
660 OpBuilderDAG<(ins "LLVMFuncOp":$func,
673 LLVMFuncOp getFunction();

12