Make sure to use the user resource when copying the input in

replace_nulls
rapidsai · Mar 30, 2021 · 0bcea32 · 0bcea32
1 parent ad9212b
commit 0bcea32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/replace/nulls.cu b/cpp/src/replace/nulls.cu
@@ -424,7 +424,7 @@ std::unique_ptr<cudf::column> replace_nulls(cudf::column_view const& input,
   CUDF_EXPECTS(replacement.size() == input.size(), "Column size mismatch");
 
   if (input.is_empty()) { return cudf::empty_like(input); }
-  if (!input.has_nulls()) { return std::make_unique<cudf::column>(input); }
+  if (!input.has_nulls()) { return std::make_unique<cudf::column>(input, stream, mr); }
 
   return cudf::type_dispatcher(
     input.type(), replace_nulls_column_kernel_forwarder{}, input, replacement, stream, mr);