add stream to parquet_chunked_writer

rapidsai · Dec 18, 2024 · dee9fda · dee9fda
1 parent 8cf0eb1
commit dee9fda
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/pylibcudf/pylibcudf/libcudf/io/parquet.pxd b/python/pylibcudf/pylibcudf/libcudf/io/parquet.pxd
@@ -239,7 +239,8 @@ cdef extern from "cudf/io/parquet.hpp" namespace "cudf::io" nogil:
     cdef cppclass parquet_chunked_writer:
         parquet_chunked_writer() except +libcudf_exception_handler
         parquet_chunked_writer(
-            chunked_parquet_writer_options args
+            chunked_parquet_writer_options args,
+            cuda_stream_view stream,
         ) except +libcudf_exception_handler
         parquet_chunked_writer& write(
             table_view table_,