Home
last modified time | relevance | path

Searched refs:hsaco (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc239 const char* hsaco = spec.cuda_cubin_in_memory().bytes(); in GetKernel() local
241 module = in_memory_modules_[hsaco]; in GetKernel()
244 if (!GpuDriver::LoadHsaco(context_, hsaco, &module)) { in GetKernel()
248 in_memory_modules_[hsaco] = module; in GetKernel()
395 bool GpuExecutor::LoadModuleFromHsaco(const char* hsaco, hipModule_t* module) { in LoadModuleFromHsaco() argument
397 std::tie(*module, module_refcount) = gpu_binary_to_module_[hsaco]; in LoadModuleFromHsaco()
400 if (!GpuDriver::LoadHsaco(context_, hsaco, module)) { in LoadModuleFromHsaco()
405 VLOG(3) << "Loaded HSACO " << static_cast<const void*>(hsaco) in LoadModuleFromHsaco()
409 VLOG(3) << "HSACO " << static_cast<const void*>(hsaco) in LoadModuleFromHsaco()
412 gpu_binary_to_module_[hsaco] = {*module, module_refcount}; in LoadModuleFromHsaco()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h282 bool LoadModuleFromHsaco(const char* hsaco, GpuModuleHandle* module)
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc262 bool GpuExecutor::LoadModuleFromHsaco(const char* hsaco, CUmodule* module) { in LoadModuleFromHsaco() argument