Skip to content

GitLab

"examples/pytorch/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "2758c249559cbd6cb7c582018a453d324913df5f"
Commit 04a2387c authored by Juha Reunanen's avatar Juha Reunanen Committed by Davis E. King
Browse files

Make sure CUDA_VERSION is defined (#1663)

parent 23f874d6
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
dlib/cuda/gpu_data.cpp
View file @ 04a2387c
...@@ -11,7 +11,7 @@ ...@@ -11,7 +11,7 @@
#include <iostream> #include <iostream>
#include "cuda_utils.h" #include "cuda_utils.h"
#include <cstring> #include <cstring>
#include <cuda.h>
namespace dlib namespace dlib
{ {
...@@ -81,7 +81,9 @@ namespace dlib ...@@ -81,7 +81,9 @@ namespace dlib
void synchronize_stream(cudaStream_t stream) void synchronize_stream(cudaStream_t stream)
{ {
#if CUDA_VERSION >= 9020 && CUDA_VERSION <= 10000 #if !defined CUDA_VERSION
#error CUDA_VERSION not defined
#elif CUDA_VERSION >= 9020 && CUDA_VERSION <= 10000
// 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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment