mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
CUDA_CHECK definition for debug
This commit is contained in:
@ -458,7 +458,7 @@ int main(int argc, char **argv)
|
||||
#else
|
||||
if(gpu_index >= 0){
|
||||
cuda_set_device(gpu_index);
|
||||
check_error(cudaSetDeviceFlags(cudaDeviceScheduleBlockingSync));
|
||||
CHECK_CUDA(cudaSetDeviceFlags(cudaDeviceScheduleBlockingSync));
|
||||
}
|
||||
#endif
|
||||
|
||||
|
Reference in New Issue
Block a user