Explicitly check the warp id

gevtushenko · May 25, 2023 · e94f1cc · e94f1cc
1 parent e1f501c
commit e94f1cc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/device/example_device_decoupled_look_back.cu b/examples/device/example_device_decoupled_look_back.cu
@@ -71,7 +71,9 @@ __global__ void decoupled_look_back_kernel(cub::ScanTileState<MessageT> tile_sta
   else
   {
     // Only the first warp in the block can perform the look back
-    if (tid < threads_in_warp)
+    const unsigned int warp_id = tid / threads_in_warp;
+
+    if (warp_id == 0)
     {
       MessageT exclusive_prefix = prefix(block_aggregate);