MCPcopy Create free account
hub / github.com/OpenPTrack/open_ptrack_v2 / cudnn_handle

Function cudnn_handle

yolo_detector/darknet_opt/src/cuda.c:65–75  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

63
64#ifdef CUDNN
65cudnnHandle_t cudnn_handle()
66{
67 static int init[16] = {0};
68 static cudnnHandle_t handle[16];
69 int i = cuda_get_device();
70 if(!init[i]) {
71 cudnnCreate(&handle[i]);
72 init[i] = 1;
73 }
74 return handle[i];
75}
76#endif
77
78cublasHandle_t blas_handle()

Callers 4

get_workspace_sizeFunction · 0.85

Calls 1

cuda_get_deviceFunction · 0.85

Tested by

no test coverage detected