NASP/cuda-samples
51
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2025-09-13 12:12:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix null pointer refrence issue with cuda driver API function pointers in case cuInit fails

Browse Source
This commit is contained in:
Nikhil Talpallikar 2025-08-01 10:27:09 -07:00
parent 8a9e2c830c
commit fd513b4846
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
Samples/0_Introduction/matrixMulDynlinkJIT/cuda_drvapi_dynlink.c
View File

@ -345,17 +345,6 @@ CUresult CUDAAPI cuInit(unsigned int Flags, int cudaVersion)
CHECKED_CALL(LOAD_LIBRARY(&CudaDrvLib));
// cuInit is required; alias it to _cuInit
GET_PROC_EX(cuInit, _cuInit, 1);
CHECKED_CALL(_cuInit(Flags));
// available since 2.2. if not present, version 1.0 is assumed
GET_PROC_OPTIONAL(cuDriverGetVersion);
if (cuDriverGetVersion) {
CHECKED_CALL(cuDriverGetVersion(&driverVer));
}
// fetch all function pointers
GET_PROC(cuDeviceGet);
GET_PROC(cuDeviceGetCount);
@ -620,5 +609,16 @@ CUresult CUDAAPI cuInit(unsigned int Flags, int cudaVersion)
#endif
}
// cuInit is required; alias it to _cuInit
GET_PROC_EX(cuInit, _cuInit, 1);
CHECKED_CALL(_cuInit(Flags));
// available since 2.2. if not present, version 1.0 is assumed
GET_PROC_OPTIONAL(cuDriverGetVersion);
if (cuDriverGetVersion) {
CHECKED_CALL(cuDriverGetVersion(&driverVer));
}
return CUDA_SUCCESS;
}