SWDEV-234029 - Remove module from module_map_ @ hipModuleUnload

Change-Id: Ic01fdb88c658c68298bd51cab2da21dc3ff0e1f7


[ROCm/clr commit: 55826ac474]
This commit is contained in:
kjayapra-amd
2020-05-04 18:59:49 -04:00
parent 3eee7bf1fc
commit 929c5a3d77
@@ -209,6 +209,7 @@ bool PlatformState::unregisterFunc(hipModule_t hmod) {
}
delete mod_ptr;
}
module_map_.erase(mod_it);
}
return true;
}
@@ -360,6 +361,7 @@ bool PlatformState::findModFunc(hipFunction_t* hfunc, hipModule_t hmod, const ch
auto mod_it = module_map_.find(hmod);
if (mod_it != module_map_.cend()) {
assert(mod_it->second != nullptr);
auto func_it = mod_it->second->functions_.find(name);
if (func_it != mod_it->second->functions_.cend()) {
PlatformState::DeviceFunction& devFunc = func_it->second;