Searched refs:module_refcount (Results 1 – 2 of 2) sorted by relevance
221 uint64_t module_refcount; in LoadModuleFromCuBin() local222 std::tie(*module, module_refcount) = gpu_binary_to_module_[cubin]; in LoadModuleFromCuBin()230 module_refcount = 1; in LoadModuleFromCuBin()234 ++module_refcount; in LoadModuleFromCuBin()238 gpu_binary_to_module_[cubin] = {*module, module_refcount}; in LoadModuleFromCuBin()243 uint64_t module_refcount; in LoadModuleFromPtx() local244 std::tie(*module, module_refcount) = gpu_binary_to_module_[ptx]; in LoadModuleFromPtx()252 module_refcount = 1; in LoadModuleFromPtx()254 ++module_refcount; in LoadModuleFromPtx()258 gpu_binary_to_module_[ptx] = {*module, module_refcount}; in LoadModuleFromPtx()
396 uint64_t module_refcount; in LoadModuleFromHsaco() local397 std::tie(*module, module_refcount) = gpu_binary_to_module_[hsaco]; in LoadModuleFromHsaco()404 module_refcount = 1; in LoadModuleFromHsaco()408 ++module_refcount; in LoadModuleFromHsaco()412 gpu_binary_to_module_[hsaco] = {*module, module_refcount}; in LoadModuleFromHsaco()