From fc4c7465d3c53bfc6403e090a217c4975b8eaf25 Mon Sep 17 00:00:00 2001 From: Jens Glaser Date: Sat, 8 Dec 2018 14:44:35 -0500 Subject: [PATCH] fix invalid arguments to warp sync error on Volta --- cub/warp/specializations/warp_scan_shfl.cuh | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/cub/warp/specializations/warp_scan_shfl.cuh b/cub/warp/specializations/warp_scan_shfl.cuh index 7f4e1c94bb..7339283302 100644 --- a/cub/warp/specializations/warp_scan_shfl.cuh +++ b/cub/warp/specializations/warp_scan_shfl.cuh @@ -107,13 +107,12 @@ struct WarpScanShfl { lane_id = LaneId(); warp_id = 0; - member_mask = 0xffffffffu >> (CUB_WARP_THREADS(PTX_ARCH) - LOGICAL_WARP_THREADS); + member_mask = 0xffffffffu; // involves all non-exited warp threads if (!IS_ARCH_WARP) { warp_id = lane_id / LOGICAL_WARP_THREADS; lane_id = lane_id % LOGICAL_WARP_THREADS; - member_mask = member_mask << (warp_id * LOGICAL_WARP_THREADS); } }