Home
last modified time | relevance | path

Searched refs:moduleOp (Results 1 – 18 of 18) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/LLVMIR/IR/
H A DFunctionCallUtils.cpp50 OpBuilder b(moduleOp.getBodyRegion()); in lookupOrCreateFn()
52 moduleOp->getLoc(), name, in lookupOrCreateFn()
57 return lookupOrCreateFn(moduleOp, kPrintI64, in lookupOrCreatePrintI64Fn()
63 return lookupOrCreateFn(moduleOp, kPrintU64, in lookupOrCreatePrintU64Fn()
69 return lookupOrCreateFn(moduleOp, kPrintF32, in lookupOrCreatePrintF32Fn()
75 return lookupOrCreateFn(moduleOp, kPrintF64, in lookupOrCreatePrintF64Fn()
103 moduleOp, kMalloc, indexType, in lookupOrCreateMallocFn()
116 moduleOp, kFree, in lookupOrCreateFreeFn()
124 moduleOp, kGenericAlloc, indexType, in lookupOrCreateGenericAllocFn()
138 moduleOp, kGenericFree, in lookupOrCreateGenericFreeFn()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/LLVMIR/
H A DFunctionCallUtils.h36 LLVM::LLVMFuncOp lookupOrCreatePrintI64Fn(ModuleOp moduleOp);
37 LLVM::LLVMFuncOp lookupOrCreatePrintU64Fn(ModuleOp moduleOp);
38 LLVM::LLVMFuncOp lookupOrCreatePrintF32Fn(ModuleOp moduleOp);
39 LLVM::LLVMFuncOp lookupOrCreatePrintF64Fn(ModuleOp moduleOp);
40 LLVM::LLVMFuncOp lookupOrCreatePrintOpenFn(ModuleOp moduleOp);
41 LLVM::LLVMFuncOp lookupOrCreatePrintCloseFn(ModuleOp moduleOp);
42 LLVM::LLVMFuncOp lookupOrCreatePrintCommaFn(ModuleOp moduleOp);
45 LLVM::LLVMFuncOp lookupOrCreateAlignedAllocFn(ModuleOp moduleOp,
47 LLVM::LLVMFuncOp lookupOrCreateFreeFn(ModuleOp moduleOp);
48 LLVM::LLVMFuncOp lookupOrCreateGenericAllocFn(ModuleOp moduleOp,
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DNormalizeMemRefs.cpp37 void normalizeFuncOpMemRefs(func::FuncOp funcOp, ModuleOp moduleOp);
41 func::FuncOp funcOp, ModuleOp moduleOp,
55 ModuleOp moduleOp = getOperation(); in runOnOperation() local
71 moduleOp.walk([&](func::FuncOp funcOp) { in runOnOperation()
80 setCalleesAndCallersNonNormalizable(funcOp, moduleOp, in runOnOperation()
90 normalizeFuncOpMemRefs(funcOp, moduleOp); in runOnOperation()
106 func::FuncOp funcOp, ModuleOp moduleOp, in setCalleesAndCallersNonNormalizable() argument
124 setCalleesAndCallersNonNormalizable(funcOp, moduleOp, in setCalleesAndCallersNonNormalizable()
137 setCalleesAndCallersNonNormalizable(funcOp, moduleOp, in setCalleesAndCallersNonNormalizable()
320 updateFunctionSignature(parentFuncOp, moduleOp); in updateFunctionSignature()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DOneShotModuleBufferize.cpp279 getFuncOpsOrderedByCalls(ModuleOp moduleOp, in getFuncOpsOrderedByCalls() argument
287 WalkResult res = moduleOp.walk([&](func::FuncOp funcOp) -> WalkResult { in getFuncOpsOrderedByCalls()
318 return moduleOp.emitOpError( in getFuncOpsOrderedByCalls()
357 mlir::bufferization::analyzeModuleOp(ModuleOp moduleOp, in analyzeModuleOp() argument
404 ModuleOp moduleOp, const OneShotAnalysisState &analysisState) { in bufferizeModuleOp() argument
409 IRRewriter rewriter(moduleOp.getContext()); in bufferizeModuleOp()
433 moduleOp.walk([&](func::FuncOp op) { in bufferizeModuleOp()
442 ModuleOp moduleOp, const OneShotBufferizationOptions &options) { in runOneShotModuleBufferize() argument
445 OneShotAnalysisState analysisState(moduleOp, options); in runOneShotModuleBufferize()
446 if (failed(insertTensorCopies(moduleOp, options))) in runOneShotModuleBufferize()
[all …]
H A DBufferUtils.cpp151 auto moduleOp = constantOp->getParentOfType<ModuleOp>(); in getGlobalFor() local
152 if (!moduleOp) in getGlobalFor()
157 for (Operation &op : moduleOp.getRegion().getOps()) { in getGlobalFor()
171 OpBuilder globalBuilder(moduleOp.getContext()); in getGlobalFor()
172 SymbolTable symbolTable(moduleOp); in getGlobalFor()
196 global->moveBefore(&moduleOp.front()); in getGlobalFor()
H A DBufferize.cpp228 ModuleOp moduleOp = getOperation(); in runOnOperation() local
230 if (failed(runOneShotModuleBufferize(moduleOp, opt))) { in runOnOperation()
235 if (failed(runOneShotBufferize(moduleOp, opt))) { in runOnOperation()
248 (void)runPipeline(cleanupPipeline, moduleOp); in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/TransformOps/
H A DBufferizationTransformOps.cpp40 auto moduleOp = dyn_cast<ModuleOp>(target); in apply() local
41 if (getTargetIsModule() && !moduleOp) in apply()
44 if (!moduleOp) in apply()
46 if (failed(bufferization::runOneShotModuleBufferize(moduleOp, options))) in apply()
/llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/
H A DGPUOpsLowering.cpp157 if (!(ret = moduleOp.template lookupSymbol<LLVM::LLVMFuncOp>(name))) { in getOrDefineFunction()
159 rewriter.setInsertionPointToStart(moduleOp.getBody()); in getOrDefineFunction()
179 auto moduleOp = gpuPrintfOp->getParentOfType<gpu::GPUModuleOp>(); in matchAndRewrite() local
182 getOrDefineFunction(moduleOp, loc, rewriter, "__ockl_printf_begin", in matchAndRewrite()
187 moduleOp, loc, rewriter, "__ockl_printf_append_args", in matchAndRewrite()
193 moduleOp, loc, rewriter, "__ockl_printf_append_string_n", in matchAndRewrite()
210 } while (moduleOp.lookupSymbol(stringConstName)); in matchAndRewrite()
220 rewriter.setInsertionPointToStart(moduleOp.getBody()); in matchAndRewrite()
297 auto moduleOp = gpuPrintfOp->getParentOfType<gpu::GPUModuleOp>(); in matchAndRewrite() local
310 } while (moduleOp.lookupSymbol(stringConstName)); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/Transforms/
H A DOneShotModuleBufferize.h23 LogicalResult analyzeModuleOp(ModuleOp moduleOp, OneShotAnalysisState &state);
27 LogicalResult bufferizeModuleOp(ModuleOp moduleOp,
35 ModuleOp moduleOp,
/llvm-project-15.0.7/mlir/lib/Conversion/GPUToSPIRV/
H A DGPUToSPIRVPass.cpp49 module.walk([&builder, &kernelModules](gpu::GPUModuleOp moduleOp) { in runOnOperation() argument
53 builder.setInsertionPoint(moduleOp.getOperation()); in runOnOperation()
54 kernelModules.push_back(builder.clone(*moduleOp.getOperation())); in runOnOperation()
H A DGPUToSPIRV.cpp85 matchAndRewrite(gpu::GPUModuleOp moduleOp, OpAdaptor adaptor,
305 gpu::GPUModuleOp moduleOp, OpAdaptor adaptor, in matchAndRewrite() argument
307 spirv::TargetEnvAttr targetEnv = spirv::lookupTargetEnvOrDefault(moduleOp); in matchAndRewrite()
311 return moduleOp.emitRemark("match failure: could not selected memory model " in matchAndRewrite()
315 std::string spvModuleName = (kSPIRVModule + moduleOp.getName()).str(); in matchAndRewrite()
317 moduleOp.getLoc(), addressingModel, *memoryModel, llvm::None, in matchAndRewrite()
322 rewriter.inlineRegionBefore(moduleOp.body(), spvModuleRegion, in matchAndRewrite()
326 rewriter.eraseOp(moduleOp); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/Transforms/
H A DUnifyAliasedResourcePass.cpp44 static AliasedResourceMap collectAliasedResources(spirv::ModuleOp moduleOp) { in collectAliasedResources() argument
46 moduleOp->walk([&aliasedResources](spirv::GlobalVariableOp varOp) { in collectAliasedResources()
211 auto moduleOp = addressOp->getParentOfType<spirv::ModuleOp>(); in shouldUnify() local
212 auto *varOp = SymbolTable::lookupSymbolIn(moduleOp, addressOp.variable()); in shouldUnify()
314 auto moduleOp = addressOp->getParentOfType<spirv::ModuleOp>(); in matchAndRewrite() local
316 SymbolTable::lookupSymbolIn(moduleOp, addressOp.variable())); in matchAndRewrite()
333 auto moduleOp = acOp->getParentOfType<spirv::ModuleOp>(); in matchAndRewrite() local
335 SymbolTable::lookupSymbolIn(moduleOp, addressOp.variable())); in matchAndRewrite()
513 spirv::ModuleOp moduleOp = getOperation(); in runOnOperation() local
530 if (failed(applyPartialConversion(moduleOp, target, std::move(patterns)))) in runOnOperation()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/Transforms/
H A DBufferizableOpInterfaceImpl.cpp39 auto moduleOp = constantOp->getParentOfType<ModuleOp>(); in bufferize() local
40 if (!moduleOp) in bufferize()
/llvm-project-15.0.7/mlir/unittests/IR/
H A DInterfaceAttachmentTest.cpp305 OwningOpRef<ModuleOp> moduleOp = in TEST() local
307 ASSERT_FALSE(isa<TestExternalOpInterface>(moduleOp->getOperation())); in TEST()
311 auto iface = dyn_cast<TestExternalOpInterface>(moduleOp->getOperation()); in TEST()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Vector/
H A DTestVectorTransforms.cpp777 ModuleOp moduleOp = warpOp->getParentOfType<ModuleOp>(); in allocateGlobalSharedMemory() local
778 SymbolTable symbolTable(moduleOp); in allocateGlobalSharedMemory()
788 builder.setInsertionPoint(moduleOp); in allocateGlobalSharedMemory()
800 global->moveBefore(&moduleOp.front()); in allocateGlobalSharedMemory()
/llvm-project-15.0.7/mlir/test/CAPI/
H A Dir.c85 MlirModule moduleOp = mlirModuleCreateEmpty(location); in makeAndDumpAdd() local
86 MlirBlock moduleBody = mlirModuleGetBody(moduleOp); in makeAndDumpAdd()
176 MlirOperation module = mlirModuleGetOperation(moduleOp); in makeAndDumpAdd()
195 return moduleOp; in makeAndDumpAdd()
485 MlirModule moduleOp = makeAndDumpAdd(ctx, location); in constructAndTraverseIr() local
486 MlirOperation module = mlirModuleGetOperation(moduleOp); in constructAndTraverseIr()
495 mlirModuleDestroy(moduleOp); in constructAndTraverseIr()
/llvm-project-15.0.7/mlir/lib/Target/Cpp/
H A DTranslateToCpp.cpp585 static LogicalResult printOperation(CppEmitter &emitter, ModuleOp moduleOp) { in printOperation() argument
588 for (Operation &op : moduleOp) { in printOperation()
/llvm-project-15.0.7/mlir/lib/Bindings/Python/
H A DIRCore.cpp2522 PyOperationRef moduleOp = PyOperation::forOperation( in populateIRCore() local
2525 PyBlock returnBlock(moduleOp, mlirModuleGetBody(self.get())); in populateIRCore()