typo fix

NVIDIA · Jan 13, 2025 · 43b2b58 · 43b2b58
1 parent 68a115f
commit 43b2b58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cub/cub/warp/specializations/warp_reduce_smem.cuh b/cub/cub/warp/specializations/warp_reduce_smem.cuh
@@ -232,7 +232,7 @@ struct WarpReduceSmem
     }
 
     // Keep bits above the current thread.
-    warp_flags &= ::cuda::ptx::get_sreg_lanemask_ge();
+    warp_flags &= ::cuda::ptx::get_sreg_lanemask_gt();
 
     // Accommodate packing of multiple logical warps in a single physical warp
     if (!IS_ARCH_WARP)