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__
|
__inline__ __device__
|
||||||
double warpReduceSum(double val) {
|
double warpReduceSum(double val) {
|
||||||
for (int offset = warpSize/2; offset > 0; offset /= 2)
|
for (int offset = warpSize/2; offset > 0; offset /= 2)
|
||||||
val += __shfl_down(val, offset);
|
val += __shfl_down_sync(0xFFFFFFFF, val, offset, 32);
|
||||||
return val;
|
return val;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user