Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
shrshi committed Aug 2, 2024
1 parent 3949cda commit 4d88fe5
Show file tree
Hide file tree
Showing 3 changed files with 128 additions and 121 deletions.
12 changes: 7 additions & 5 deletions cpp/src/io/json/column_tree_construction.cu
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,8 @@ std::tuple<csr, column_tree_properties> reduce_to_column_tree(
auto* dev_num_levels_ptr = thrust::max_element(
rmm::exec_policy_nosync(stream), tree.node_levels.begin(), tree.node_levels.end());
rmm::device_scalar<NodeIndexT> num_levels(stream);
CUDF_CUDA_TRY(cudaMemcpyAsync(num_levels.data(), dev_num_levels_ptr, sizeof(NodeIndexT), cudaMemcpyDeviceToDevice, stream));
CUDF_CUDA_TRY(cudaMemcpyAsync(
num_levels.data(), dev_num_levels_ptr, sizeof(NodeIndexT), cudaMemcpyDeviceToDevice, stream));

rmm::device_uvector<NodeIndexT> mapped_col_ids_copy(num_columns, stream);
thrust::copy(rmm::exec_policy_nosync(stream),
Expand Down Expand Up @@ -335,10 +336,11 @@ std::tuple<csr, column_tree_properties> reduce_to_column_tree(
[] __device__(auto ancestor) { return ancestor != -1; });
}

return std::tuple{
csr{std::move(rowidx), std::move(colidx)},
column_tree_properties{std::move(num_levels),
std::move(column_categories), std::move(max_row_offsets), std::move(mapped_col_ids)}};
return std::tuple{csr{std::move(rowidx), std::move(colidx)},
column_tree_properties{std::move(num_levels),
std::move(column_categories),
std::move(max_row_offsets),
std::move(mapped_col_ids)}};
}

} // namespace experimental::detail
Expand Down
Loading

0 comments on commit 4d88fe5

Please sign in to comment.