Change the mask scan to make the failing test pass

magnatelee · Feb 23, 2021 · c7f296e · c7f296e
1 parent 58b0185
commit c7f296e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cpp/src/reductions/scan.cu b/cpp/src/reductions/scan.cu
@@ -116,8 +116,9 @@ struct scan_dispatcher {
       thrust::find_if_not(
         rmm::exec_policy(stream), v, v + input_view.size(), thrust::identity<bool>{}) -
       v;
-    cudf::set_null_mask(
-      static_cast<cudf::bitmask_type*>(mask.data()), 0, first_null_position, true);
+    if (first_null_position > 0)
+      cudf::set_null_mask(
+        static_cast<cudf::bitmask_type*>(mask.data()), 0, first_null_position, true);
     cudf::set_null_mask(
       static_cast<cudf::bitmask_type*>(mask.data()), first_null_position, input_view.size(), false);
     return mask;