replace deprecated shfl_down
This commit is contained in:
parent
ab70a9aeba
commit
11a4a4b848
@ -89,7 +89,7 @@ __global__ void sum_kernel_block(double *sum, double *input, int n)
|
||||
__inline__ __device__
|
||||
double warpReduceSum(double val) {
|
||||
for (int offset = warpSize/2; offset > 0; offset /= 2)
|
||||
val += __shfl_down(val, offset);
|
||||
val += __shfl_down_sync(0xFFFFFFFF, val, offset, 32);
|
||||
return val;
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user