Searched refs:module_refcount (Results 1 – 2 of 2) sorted by relevance
222 uint64_t module_refcount; in LoadModuleFromCuBin() local223 std::tie(*module, module_refcount) = gpu_binary_to_module_[cubin]; in LoadModuleFromCuBin()227 module_refcount = 1; in LoadModuleFromCuBin()231 ++module_refcount; in LoadModuleFromCuBin()235 gpu_binary_to_module_[cubin] = {*module, module_refcount}; in LoadModuleFromCuBin()240 uint64_t module_refcount; in LoadModuleFromPtx() local241 std::tie(*module, module_refcount) = gpu_binary_to_module_[ptx]; in LoadModuleFromPtx()247 module_refcount = 1; in LoadModuleFromPtx()249 ++module_refcount; in LoadModuleFromPtx()253 gpu_binary_to_module_[ptx] = {*module, module_refcount}; in LoadModuleFromPtx()
398 uint64_t module_refcount; in LoadModuleFromHsaco() local399 std::tie(*module, module_refcount) = gpu_binary_to_module_[hsaco]; in LoadModuleFromHsaco()403 module_refcount = 1; in LoadModuleFromHsaco()408 ++module_refcount; in LoadModuleFromHsaco()412 gpu_binary_to_module_[hsaco] = {*module, module_refcount}; in LoadModuleFromHsaco()