OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix macro

This commit is contained in:
cyy 2020-01-10 17:47:10 +08:00
parent 8bd01dd8da
commit 300406e26a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/dark_cuda.c
View File

@ -162,7 +162,7 @@ cudnnHandle_t cudnn_handle()
cudnnCreate(&handle[i]);
init[i] = 1;
cudnnStatus_t status = cudnnSetStream(handle[i], get_cuda_stream());
CHECK_CUDA(status);
CHECK_CUDNN(status);
}
return handle[i];
}