From 0bcea328994cdf20a1d124e55b13851e77022a51 Mon Sep 17 00:00:00 2001 From: Wonchan Lee Date: Tue, 30 Mar 2021 14:56:25 -0700 Subject: [PATCH] Make sure to use the user resource when copying the input in replace_nulls --- cpp/src/replace/nulls.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/src/replace/nulls.cu b/cpp/src/replace/nulls.cu index afc2bbb37bd..b5418c079d9 100644 --- a/cpp/src/replace/nulls.cu +++ b/cpp/src/replace/nulls.cu @@ -424,7 +424,7 @@ std::unique_ptr 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(input); } + if (!input.has_nulls()) { return std::make_unique(input, stream, mr); } return cudf::type_dispatcher( input.type(), replace_nulls_column_kernel_forwarder{}, input, replacement, stream, mr);