Searched refs:kernelModule (Results 1 – 5 of 5) sorted by relevance
257 auto kernelModule = createKernelModule(outlinedFunc, symbolTable); in runOnOperation() local258 symbolTable.insert(kernelModule, insertPt); in runOnOperation()289 auto kernelModule = cast<gpu::GPUModuleOp>(Operation::create(state)); in createKernelModule() local290 SymbolTable symbolTable(kernelModule); in createKernelModule()311 return kernelModule; in createKernelModule()
58 std::unique_ptr<llvm::Module> kernelModule = in convertMLIRModule() local64 llvm::Linker::linkModules(*mainModule, std::move(kernelModule)); in convertMLIRModule()
272 auto kernelModule = ::mlir::ModuleOp::create(module.getLoc()); in ExtractKernelModule() local274 module.walk([&kernelModule](mlir::gpu::GPUModuleOp nestedModule) { in ExtractKernelModule()276 kernelModule.push_back(fn.clone()); in ExtractKernelModule()279 return kernelModule; in ExtractKernelModule()
586 auto kernelModule = SymbolTable::lookupNearestSymbolFrom<gpu::GPUModuleOp>( in matchAndRewrite() local588 assert(kernelModule && "expected a kernel module"); in matchAndRewrite()591 kernelModule->getAttrOfType<StringAttr>(gpuBinaryAnnotation); in matchAndRewrite()593 kernelModule.emitOpError() in matchAndRewrite()598 SmallString<128> nameBuffer(kernelModule.getName()); in matchAndRewrite()
95 auto kernelModule = module.lookupSymbol<GPUModuleOp>(kernelModuleName); in verifyOperationAttribute() local96 if (!kernelModule) in verifyOperationAttribute()437 auto kernelModule = kernelFunc->getParentOfType<GPUModuleOp>(); in build() local439 kernelModule.getName(), {builder.getSymbolRefAttr(kernelFunc.getName())}); in build()