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
__inline__ __device__ float warp_reduce_sum(float val)
{
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);
#endif
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