diff --git a/cpp/src/dictionary/detail/merge.cu b/cpp/src/dictionary/detail/merge.cu index 2ff0a3e0a2a..e972403cad3 100644 --- a/cpp/src/dictionary/detail/merge.cu +++ b/cpp/src/dictionary/detail/merge.cu @@ -17,6 +17,7 @@ #include #include #include +#include #include #include #include @@ -62,8 +63,11 @@ std::unique_ptr merge(dictionary_column_view const& lcol, return make_dictionary_column( std::make_unique(lcol.keys(), stream, mr), std::move(indices_column), - rmm::device_buffer{ - lcol.has_nulls() || rcol.has_nulls() ? static_cast(merged_size) : 0, stream, mr}, + cudf::detail::create_null_mask( + lcol.has_nulls() || rcol.has_nulls() ? static_cast(merged_size) : 0, + mask_state::UNINITIALIZED, + stream, + mr), lcol.null_count() + rcol.null_count()); }