From 7aaf5eddd27e881f124fb47c13d4d4e1d33299b0 Mon Sep 17 00:00:00 2001 From: Vukasin Milovanovic Date: Mon, 23 Dec 2024 15:56:00 -0800 Subject: [PATCH] fix --- cpp/src/io/comp/comp.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cpp/src/io/comp/comp.cpp b/cpp/src/io/comp/comp.cpp index e61aefe680e..63bc697b33f 100644 --- a/cpp/src/io/comp/comp.cpp +++ b/cpp/src/io/comp/comp.cpp @@ -125,6 +125,8 @@ void device_compress(compression_type compression, device_span results, rmm::cuda_stream_view stream) { + if (compression == compression_type::NONE) { return; } + auto const nvcomp_type = to_nvcomp_compression(compression); auto nvcomp_disabled = nvcomp_type.has_value() ? nvcomp::is_compression_disabled(*nvcomp_type) : "invalid compression type"; @@ -144,8 +146,8 @@ void host_compress(compression_type compression, device_span results, rmm::cuda_stream_view stream) { - CUDF_FUNC_RANGE(); if (compression == compression_type::NONE) { return; } + auto const num_blocks = inputs.size(); auto h_results = cudf::detail::make_host_vector(num_blocks, stream); auto const h_inputs = cudf::detail::make_host_vector_async(inputs, stream);