Adding checks for return type compatibility with CUDA

Change-Id: I09b6bd9a7299f0a97ac9011ea0903a1f81e28251


[ROCm/hip commit: a779350181]
This commit is contained in:
Jatin Chaudhary
2020-09-16 08:27:55 -04:00
committed by Jatin Chaudhary
parent bc1f8180ee
commit 42636e83e5
2 changed files with 6 additions and 0 deletions
+3
View File
@@ -64,6 +64,9 @@ hipError_t FatBinaryInfo::ExtractFatBinary(const std::vector<hip::Device*>& devi
if (!amd::Os::GetFileHandle(fname_.c_str(), &fdesc_, &fsize_)) {
return hipErrorFileNotFound;
}
if (fsize_ == 0) {
return hipErrorInvalidKernelFile;
}
// Extract the code object from file
hip_error = CodeObject::ExtractCodeObjectFromFile(fdesc_, fsize_,
+3
View File
@@ -773,6 +773,9 @@ hipError_t PlatformState::getDynFunc(hipFunction_t* hfunc, hipModule_t hmod,
DevLogPrintfError("Cannot find the module: 0x%x", hmod);
return hipErrorNotFound;
}
if (0 == strlen(func_name)) {
return hipErrorNotFound;
}
return it->second->getDynFunc(hfunc, func_name);
}