Merge pull request #319 from senior-zero/main-bug/github/find_heads_a…

…nd_tails_oob/314 Fix OOB access in FlagHeadsAndTails
NVIDIA · Jun 14, 2021 · 0c7d105 · 0c7d105
2 parents e3ee09f + 9d8e886
commit 0c7d105
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cub/block/block_discontinuity.cuh b/cub/block/block_discontinuity.cuh
@@ -747,13 +747,13 @@ public:
         T preds[ITEMS_PER_THREAD];
 
         // Set flag for first thread-item
-        preds[0] = temp_storage.last_items[linear_tid - 1];
         if (linear_tid == 0)
         {
             head_flags[0] = 1;
         }
         else
         {
+            preds[0] = temp_storage.last_items[linear_tid - 1];
             head_flags[0] = ApplyOp<FlagOp>::FlagT(
                 flag_op,
                 preds[0],