forked from rapidsai/cudf
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathfinalize_partitioned_full_join.cpp
More file actions
35 lines (29 loc) · 1.06 KB
/
finalize_partitioned_full_join.cpp
File metadata and controls
35 lines (29 loc) · 1.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
/*
* SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION.
* SPDX-License-Identifier: Apache-2.0
*/
#include "join/join_common_utils.hpp"
#include <cudf/detail/nvtx/ranges.hpp>
#include <cudf/join/hash_join.hpp>
#include <cudf/join/join.hpp>
#include <cudf/types.hpp>
#include <cudf/utilities/memory_resource.hpp>
#include <cudf/utilities/span.hpp>
#include <rmm/cuda_stream_view.hpp>
#include <rmm/device_uvector.hpp>
namespace cudf {
std::pair<std::unique_ptr<rmm::device_uvector<size_type>>,
std::unique_ptr<rmm::device_uvector<size_type>>>
hash_join::finalize_partitioned_full_join(
cudf::host_span<cudf::device_span<size_type const> const> left_partials,
cudf::host_span<cudf::device_span<size_type const> const> right_partials,
size_type left_table_num_rows,
size_type right_table_num_rows,
rmm::cuda_stream_view stream,
rmm::device_async_resource_ref mr)
{
CUDF_FUNC_RANGE();
return cudf::detail::finalize_full_join(
left_partials, right_partials, left_table_num_rows, right_table_num_rows, stream, mr);
}
} // namespace cudf