use span

rapidsai · Oct 31, 2022 · 8cf2b5a · 8cf2b5a
1 parent 77e32f2
commit 8cf2b5a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cpp/include/cudf/io/detail/csv.hpp b/cpp/include/cudf/io/detail/csv.hpp
@@ -53,7 +53,7 @@ table_with_metadata read_csv(std::unique_ptr<cudf::io::datasource>&& source,
  */
 void write_csv(data_sink* sink,
                table_view const& table,
-               std::vector<std::string> const& column_names,
+               host_span<std::string const> column_names,
                csv_writer_options const& options,
                rmm::cuda_stream_view stream,
                rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());

diff --git a/cpp/src/io/csv/writer_impl.cu b/cpp/src/io/csv/writer_impl.cu
@@ -279,7 +279,7 @@ struct column_to_strings_fn {
 //
 void write_chunked_begin(data_sink* out_sink,
                          table_view const& table,
-                         std::vector<std::string> const& user_column_names,
+                         host_span<std::string const> user_column_names,
                          csv_writer_options const& options,
                          rmm::cuda_stream_view stream,
                          rmm::mr::device_memory_resource* mr)
@@ -398,7 +398,7 @@ void write_chunked(data_sink* out_sink,
 
 void write_csv(data_sink* out_sink,
                table_view const& table,
-               std::vector<std::string> const& user_column_names,
+               host_span<std::string const> user_column_names,
                csv_writer_options const& options,
                rmm::cuda_stream_view stream,
                rmm::mr::device_memory_resource* mr)