Searched refs:gpu_module (Results 1 – 2 of 2) sorted by relevance
67 mlir::gpu::GPUModuleOp gpu_module = getOperation(); in runOnOperation() local68 auto blob_or = GetGpuBinaryBlob(gpu_module); in runOnOperation()72 gpu_module->setAttr(blob_annotation_, in runOnOperation()77 gpu_module.emitError(blob_or.status().error_message()); in runOnOperation()82 mlir::gpu::GPUModuleOp gpu_module) { in GetGpuBinaryBlob() argument93 auto llvmModule = mlir::translateModuleToLLVMIR(gpu_module, llvmContext); in GetGpuBinaryBlob()196 << gpu_module.getName() << "' on architecture sm_" << arch in GetGpuBinaryBlob()
253 auto gpu_module = kernel->getParentOfType<mlir::gpu::GPUModuleOp>(); in runOnFunction() local254 mlir::OpBuilder kernel_builder(gpu_module.body()); in runOnFunction()