SWDEV-236178 - Store texture reference metadata for dynamically loaded modules.

Change-Id: I99ecc80da7e29c691341a01a09e4532972f1e3e5

rocclr/hip_platform.cpp

@@ -955,6 +955,15 @@ hipError_t PlatformState::unloadModule(hipModule_t hmod) {
   delete it->second;
   dynCO_map_.erase(hmod);
 
+  auto tex_it = texRef_map_.begin();
+  while (tex_it != texRef_map_.end()) {
+    if (tex_it->second.first == hmod) {
+      tex_it = texRef_map_.erase(tex_it);
+    } else {
+      ++tex_it;
+    }
+  }
+
   return hipSuccess;
 }
 
@@ -989,6 +998,37 @@ hipError_t PlatformState::getDynGlobalVar(const char* hostVar, int deviceId, hip
   return hipSuccess;
 }
 
+hipError_t PlatformState::registerTexRef(textureReference* texRef, hipModule_t hmod,
+                                         std::string name) {
+  amd::ScopedLock lock(lock_);
+  texRef_map_.insert(std::make_pair(texRef, std::make_pair(hmod, name)));
+  return hipSuccess;
+}
+
+hipError_t PlatformState::getDynTexGlobalVar(textureReference* texRef, int deviceId,
+                                             hipDeviceptr_t* dev_ptr, size_t* size_ptr) {
+  amd::ScopedLock lock(lock_);
+
+  auto tex_it = texRef_map_.find(texRef);
+  if (tex_it == texRef_map_.end()) {
+    DevLogPrintfError("Cannot find the texRef Entry: 0x%x", texRef);
+    return hipErrorNotFound;
+  }
+
+  auto it = dynCO_map_.find(tex_it->second.first);
+  if (it == dynCO_map_.end()) {
+    DevLogPrintfError("Cannot find the module: 0x%x", tex_it->second.first);
+    return hipErrorNotFound;
+  }
+
+  hip::DeviceVar* dvar = nullptr;
+  IHIP_RETURN_ONFAIL(it->second->getDeviceVar(&dvar, tex_it->second.second, deviceId));
+  *dev_ptr = dvar->device_ptr();
+  *size_ptr = dvar->size();
+
+  return hipSuccess;
+}
+
 hipError_t PlatformState::getDynTexRef(const char* hostVar, hipModule_t hmod, textureReference** texRef) {
   amd::ScopedLock lock(lock_);