NASP/cuda-samples
35
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2024-11-24 14:39:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

add check for filename in nvrtc_helper.h

Browse Source
This commit is contained in:
Rutwik Choughule 2022-02-03 18:12:24 +05:30
parent 8f21b899b6
commit b312abaa07
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Common/helper_string.h
View File

@ -421,6 +421,7 @@ inline char *sdkFindFilePath(const char *filename,
} }
// File not found // File not found
printf("\nerror: sdkFindFilePath: file <%s> not found!\n", filename);
return 0; return 0;
} }

13
Common/nvrtc_helper.h
View File

@ -49,6 +49,11 @@
void compileFileToCUBIN(char *filename, int argc, char **argv, char **cubinResult, void compileFileToCUBIN(char *filename, int argc, char **argv, char **cubinResult,
size_t *cubinResultSize, int requiresCGheaders) { size_t *cubinResultSize, int requiresCGheaders) {
if (!filename) {
std::cerr << "\nerror: filename is empty for compileFileToCUBIN()!\n";
exit(1);
}
std::ifstream inputFile(filename, std::ifstream inputFile(filename,
std::ios::in | std::ios::binary | std::ios::ate); std::ios::in | std::ios::binary | std::ios::ate);
@ -111,7 +116,12 @@ void compileFileToCUBIN(char *filename, int argc, char **argv, char **cubinResul
compileOptions = "--include-path="; compileOptions = "--include-path=";
std::string path = sdkFindFilePath(HeaderNames, argv[0]); char *strPath = sdkFindFilePath(HeaderNames, argv[0]);
if (!strPath) {
std::cerr << "\nerror: header file " << HeaderNames << " not found!\n";
exit(1);
}
std::string path = strPath;
if (!path.empty()) { if (!path.empty()) {
std::size_t found = path.find(HeaderNames); std::size_t found = path.find(HeaderNames);
path.erase(found); path.erase(found);
@ -120,6 +130,7 @@ void compileFileToCUBIN(char *filename, int argc, char **argv, char **cubinResul
"\nCooperativeGroups headers not found, please install it in %s " "\nCooperativeGroups headers not found, please install it in %s "
"sample directory..\n Exiting..\n", "sample directory..\n Exiting..\n",
argv[0]); argv[0]);
exit(1);
} }
compileOptions += path.c_str(); compileOptions += path.c_str();
compileParams[numCompileOptions] = reinterpret_cast<char *>( compileParams[numCompileOptions] = reinterpret_cast<char *>(