From c3835315437ef69e2542499bde146b5326ea46bd Mon Sep 17 00:00:00 2001 From: JayjeetAtGithub Date: Tue, 9 Jul 2024 01:07:34 -0700 Subject: [PATCH] Remove mr param from write_chunked --- cpp/src/io/json/write_json.cu | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/cpp/src/io/json/write_json.cu b/cpp/src/io/json/write_json.cu index 92f18b57321..c688c809e04 100644 --- a/cpp/src/io/json/write_json.cu +++ b/cpp/src/io/json/write_json.cu @@ -805,8 +805,7 @@ void write_chunked(data_sink* out_sink, strings_column_view const& str_column_view, int const skip_last_chars, json_writer_options const& options, - rmm::cuda_stream_view stream, - rmm::device_async_resource_ref mr) + rmm::cuda_stream_view stream) { CUDF_FUNC_RANGE(); CUDF_EXPECTS(str_column_view.size() > 0, "Unexpected empty strings column."); @@ -911,12 +910,7 @@ void write_json(data_sink* out_sink, bool const include_line_terminator = (&sub_view != &vector_views.back()) or options.is_enabled_lines(); auto const skip_last_chars = (include_line_terminator ? 0 : line_terminator.size()); - write_chunked(out_sink, - str_concat_col->view(), - skip_last_chars, - options, - stream, - rmm::mr::get_current_device_resource()); + write_chunked(out_sink, str_concat_col->view(), skip_last_chars, options, stream); } } else { if (options.is_enabled_lines()) {