Skip to content

Commit

Permalink
Add optimized 2x string column renumbering code (#1996)
Browse files Browse the repository at this point in the history
  • Loading branch information
chirayuG-nvidia authored Jan 26, 2022
1 parent 63efaea commit e0038f0
Show file tree
Hide file tree
Showing 7 changed files with 1,995 additions and 6 deletions.
6 changes: 5 additions & 1 deletion cpp/libcugraph_etl/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#=============================================================================
# Copyright (c) 2021, NVIDIA CORPORATION.
# Copyright (c) 2021-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.
Expand Down Expand Up @@ -119,6 +119,10 @@ target_compile_options(cugraph_etl
PRIVATE "$<$<COMPILE_LANGUAGE:CXX>:${CUGRAPH_ETL_CXX_FLAGS}>"
)

target_compile_options(cugraph_etl PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:
--expt-relaxed-constexpr
--expt-extended-lambda
>)
################################################################################
# - ETL include paths --------------------------------------------------------
target_include_directories(cugraph_etl
Expand Down
7 changes: 5 additions & 2 deletions cpp/libcugraph_etl/include/cugraph_etl/functions.hpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright (c) 2021, NVIDIA CORPORATION.
* Copyright (c) 2021-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.
Expand All @@ -19,6 +19,8 @@
#include <cudf/table/table.hpp>
#include <cudf/types.hpp>

#include <raft/handle.hpp>

namespace cugraph {
namespace etl {

Expand Down Expand Up @@ -47,7 +49,8 @@ namespace etl {
*/
std::
tuple<std::unique_ptr<cudf::column>, std::unique_ptr<cudf::column>, std::unique_ptr<cudf::table>>
renumber_cudf_tables(cudf::table_view const& src_table,
renumber_cudf_tables(raft::handle_t const& handle,
cudf::table_view const& src_table,
cudf::table_view const& dst_table,
cudf::type_id dtype);

Expand Down
Loading

0 comments on commit e0038f0

Please sign in to comment.