Searched refs:kernelModule (Results 1 – 4 of 4) sorted by relevance
314 auto kernelModule = createKernelModule(outlinedFunc, symbolTable); in runOnOperation() local315 symbolTable.insert(kernelModule, insertPt); in runOnOperation()343 auto kernelModule = builder.create<gpu::GPUModuleOp>(kernelFunc.getLoc(), in createKernelModule() local349 kernelModule->setAttr(DLTIDialect::kDataLayoutAttrName, dataLayoutSpec); in createKernelModule()351 SymbolTable symbolTable(kernelModule); in createKernelModule()372 return kernelModule; in createKernelModule()
64 std::unique_ptr<llvm::Module> kernelModule = in convertMLIRModule() local70 llvm::Linker::linkModules(*mainModule, std::move(kernelModule)); in convertMLIRModule()
736 auto kernelModule = SymbolTable::lookupNearestSymbolFrom<gpu::GPUModuleOp>( in matchAndRewrite() local738 assert(kernelModule && "expected a kernel module"); in matchAndRewrite()741 kernelModule->getAttrOfType<StringAttr>(gpuBinaryAnnotation); in matchAndRewrite()743 kernelModule.emitOpError() in matchAndRewrite()748 SmallString<128> nameBuffer(kernelModule.getName()); in matchAndRewrite()
221 auto kernelModule = module.lookupSymbol<GPUModuleOp>(kernelModuleName); in verifyOperationAttribute() local222 if (!kernelModule) in verifyOperationAttribute()702 auto kernelModule = kernelFunc->getParentOfType<GPUModuleOp>(); in build() local704 SymbolRefAttr::get(kernelModule.getNameAttr(), in build()