AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-255979 - Add support for dynamic __managed__ variables

Browse Source 
Change-Id: I62b790853ea3ab3b7ac57bab389046c627fdecce
This commit is contained in:
Anusha GodavarthySurya
2021-03-16 23:54:10 -07:00
committed by Anusha Godavarthy Surya
parent eb2c98bb00
commit d9fffacfb3
6 changed files with 197 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rocclr/hip_platform.cpp
+9 -6
View File
@@ -820,12 +820,15 @@ hipError_t PlatformState::getDynGlobalVar(const char* hostVar, hipModule_t hmod,
LogPrintfError("Cannot find the module: 0x%x", hmod);
return hipErrorNotFound;
}
hip::DeviceVar* dvar = nullptr;
IHIP_RETURN_ONFAIL(it->second->getDeviceVar(&dvar, hostVar));
*dev_ptr = dvar->device_ptr();
*size_ptr = dvar->size();
*dev_ptr = nullptr;
it->second->getManagedVarPointer(hostVar, dev_ptr, size_ptr);
// if dev_ptr is nullptr, hostvar is not in managed variable list
if (*dev_ptr == nullptr) {
hip::DeviceVar* dvar = nullptr;
IHIP_RETURN_ONFAIL(it->second->getDeviceVar(&dvar, hostVar));
*dev_ptr = dvar->device_ptr();
*size_ptr = dvar->size();
}
return hipSuccess;
}