Searched refs:gpu_binary (Results 1 – 4 of 4) sorted by relevance
113 const char* gpu_binary = reinterpret_cast<const char*>(module_handle.id()); in UnloadModule() local115 return UnloadGpuBinary(gpu_binary); in UnloadModule()124 bool GpuExecutor::UnloadGpuBinary(const void* gpu_binary) { in UnloadGpuBinary() argument125 auto module_it = gpu_binary_to_module_.find(gpu_binary); in UnloadGpuBinary()127 VLOG(3) << "No loaded HSACO module for " << gpu_binary; in UnloadGpuBinary()
295 bool GpuExecutor::UnloadGpuBinary(const void* gpu_binary) { in UnloadGpuBinary() argument296 auto module_it = gpu_binary_to_module_.find(gpu_binary); in UnloadGpuBinary()298 VLOG(3) << "No loaded CUDA module for " << gpu_binary; in UnloadGpuBinary()361 const char* gpu_binary = reinterpret_cast<const char*>(module_handle.id()); in UnloadModule() local363 return UnloadGpuBinary(gpu_binary); in UnloadModule()
358 bool UnloadGpuBinary(const void* gpu_binary)
374 auto gpu_binary = gpu_module->getAttrOfType<mlir::StringAttr>("binary"); in matchAndRewrite() local378 call->setAttr(b.getStringAttr("ptx"), gpu_binary); in matchAndRewrite()