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

Clean implementation for failure path when cuInit fails

Browse Source
This commit is contained in:
Nikhil Talpallikar 2025-08-05 13:46:44 -07:00
parent fd513b4846
commit f8aab0053f
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -342,9 +342,21 @@ CUresult CUDAAPI cuInit(unsigned int Flags, int cudaVersion)
{ {
CUDADRIVER CudaDrvLib; CUDADRIVER CudaDrvLib;
int driverVer = 1000; int driverVer = 1000;
CUresult result = CUDA_SUCCESS;
CHECKED_CALL(LOAD_LIBRARY(&CudaDrvLib)); result = LOAD_LIBRARY(&CudaDrvLib);
// cuInit is required; alias it to _cuInit
GET_PROC_EX(cuInit, _cuInit, 1);
result = _cuInit(Flags);
// available since 2.2. if not present, version 1.0 is assumed
GET_PROC_OPTIONAL(cuDriverGetVersion);
if (cuDriverGetVersion) {
result = cuDriverGetVersion(&driverVer);
}
// fetch all function pointers // fetch all function pointers
GET_PROC(cuDeviceGet); GET_PROC(cuDeviceGet);
GET_PROC(cuDeviceGetCount); GET_PROC(cuDeviceGetCount);
@ -608,17 +620,5 @@ CUresult CUDAAPI cuInit(unsigned int Flags, int cudaVersion)
GET_PROC(cuGraphicsD3D9RegisterResource); GET_PROC(cuGraphicsD3D9RegisterResource);
#endif #endif
} }
return result;
// 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;
} }