SWDEV-446959 - Check fatbinaryinfo for nullptr and return hipErrorInvalidDevice.

Change-Id: Iabc0d8b66ddf73a92e67dd469f1a8bc0b54b9309
2024-02-23 14:45:43 -06:00
parent e2d2fad56c
commit 8a715ba58c
1 changed files with 4 additions and 1 deletions
@@ -167,7 +167,10 @@ hipError_t Function::getStatFunc(hipFunction_t* hfunc, int deviceId) {
 }

 hipError_t Function::getStatFuncAttr(hipFuncAttributes* func_attr, int deviceId) {
-  guarantee((modules_ != nullptr), "Module not initialized");
+
+  if (modules_ == nullptr || *modules_ == nullptr) {
+    return hipErrorInvalidDeviceFunction;
+  }

  hipModule_t hmod = nullptr;
  IHIP_RETURN_ONFAIL((*modules_)->BuildProgram(deviceId));