Change CUDA version check for cudaSync from 10000 to 10010 (#1704)

64ad6b25 · Facundo Galán · Davis E. King · ae406bf4 · 64ad6b25
Commit 64ad6b25 authored Mar 27, 2019 by Facundo Galán Committed by Davis E. King Mar 27, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

dlib/cuda/gpu_data.cpp dlib/cuda/gpu_data.cpp +1 -1

No files found.
--- a/dlib/cuda/gpu_data.cpp
+++ b/dlib/cuda/gpu_data.cpp
@@ -83,7 +83,7 @@ namespace dlib
    {
 #if !defined CUDA_VERSION
 #error CUDA_VERSION not defined
-#elif CUDA_VERSION >= 9020 && CUDA_VERSION <= 10000
+#elif CUDA_VERSION >= 9020 && CUDA_VERSION <= 10010
        // This should be pretty much the same as cudaStreamSynchronize, which for some
        // reason makes training freeze in some cases.
        // (see https://github.com/davisking/dlib/issues/1513)