From 370b41f008d190e0402ad059857ef9dfda474265 Mon Sep 17 00:00:00 2001 From: sperlingxx Date: Fri, 11 Mar 2022 14:59:02 +0800 Subject: [PATCH 1/2] clean up null mask after purging null entries Signed-off-by: sperlingxx --- cpp/src/groupby/sort/group_collect.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/cpp/src/groupby/sort/group_collect.cu b/cpp/src/groupby/sort/group_collect.cu index 43a1674d97f..14120a10209 100644 --- a/cpp/src/groupby/sort/group_collect.cu +++ b/cpp/src/groupby/sort/group_collect.cu @@ -57,6 +57,7 @@ std::pair, std::unique_ptr> purge_null_entries( cudf::detail::copy_if(table_view{{values}}, not_null_pred, stream, mr)->release(); auto null_purged_values = std::move(null_purged_entries.front()); + null_purged_values->set_null_mask(rmm::device_buffer{0, stream, mr}, 0); // Recalculate offsets after null entries are purged. rmm::device_uvector null_purged_sizes(num_groups, stream); From 2a361687df6b000441eaa118797d4454cf1455ec Mon Sep 17 00:00:00 2001 From: sperlingxx Date: Fri, 11 Mar 2022 15:14:59 +0800 Subject: [PATCH 2/2] update copyright header Signed-off-by: sperlingxx --- cpp/src/groupby/sort/group_collect.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/src/groupby/sort/group_collect.cu b/cpp/src/groupby/sort/group_collect.cu index 14120a10209..52cf4fe3bff 100644 --- a/cpp/src/groupby/sort/group_collect.cu +++ b/cpp/src/groupby/sort/group_collect.cu @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2021, NVIDIA CORPORATION. + * Copyright (c) 2020-2022, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License.