Fix incorrect shfl_xor for Windows
copy/paste error, need __shfl_xor w/ lane_mask
[ROCm/hip commit: 09edc7e49c]
This commit is contained in:
committed by
GitHub
orang tua
468dccb725
melakukan
98b7cb62aa
@@ -557,7 +557,7 @@ long __shfl_xor(long var, int lane_mask, int width = warpSize)
|
||||
return tmp1;
|
||||
#else
|
||||
static_assert(sizeof(long) == sizeof(int), "");
|
||||
return static_cast<long>(__shfl_down(static_cast<int>(var), lane_delta, width));
|
||||
return static_cast<long>(__shfl_xor(static_cast<int>(var), lane_mask, width));
|
||||
#endif
|
||||
}
|
||||
__device__
|
||||
|
||||
Reference in New Issue
Block a user