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

Even the newest CUDA runtime has a buggy cudaStreamSynchronize.

This commit is contained in:
Davis King 2020-01-05 08:17:25 -05:00
parent 3d5a3d7b9a
commit 5557577c95
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
dlib/cuda/gpu_data.cpp
View File

@ -83,7 +83,10 @@ namespace dlib
{ {
#if !defined CUDA_VERSION #if !defined CUDA_VERSION
#error CUDA_VERSION not defined #error CUDA_VERSION not defined
#elif CUDA_VERSION >= 9020 && CUDA_VERSION <= 10010 #elif CUDA_VERSION >= 9020 && CUDA_VERSION <= 11000
// We will stop using this alternative version with cuda V11, hopefully the bug in
// cudaStreamSynchronize is fixed by then.
//
// This should be pretty much the same as cudaStreamSynchronize, which for some // This should be pretty much the same as cudaStreamSynchronize, which for some
// reason makes training freeze in some cases. // reason makes training freeze in some cases.
// (see https://github.com/davisking/dlib/issues/1513) // (see https://github.com/davisking/dlib/issues/1513)