Commit 176e329d authored by Davis King's avatar Davis King

Upgrade to the newer shuffle command to avoid warnings about deprecation from CUDA 9.0.

parent caea9a2c
...@@ -108,7 +108,11 @@ namespace dlib ...@@ -108,7 +108,11 @@ namespace dlib
__inline__ __device__ float warp_reduce_sum(float val) __inline__ __device__ float warp_reduce_sum(float val)
{ {
for (int offset = warpSize/2; offset > 0; offset /= 2) for (int offset = warpSize/2; offset > 0; offset /= 2)
#if CUDART_VERSION >= 9000
val += __shfl_down_sync(0xFFFFFFFF,val, offset);
#else
val += __shfl_down(val, offset); val += __shfl_down(val, offset);
#endif
return val; return val;
} }
......
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