diff --git a/cpp/src/io/parquet/decode_fixed.cu b/cpp/src/io/parquet/decode_fixed.cu index 20f560d19ed..ab2c3fe27ef 100644 --- a/cpp/src/io/parquet/decode_fixed.cu +++ b/cpp/src/io/parquet/decode_fixed.cu @@ -321,7 +321,11 @@ __global__ void __launch_bounds__(decode_block_size) gpuDecodePageDataFixed( // TODO: abellina all_types_filter??? //if (!setupLocalPageInfo(s, pp, chunks, min_row, num_rows, all_types_filter{}, true)) { return; } - if (!setupLocalPageInfo(s, pp, chunks, min_row, num_rows, mask_filter{decode_kernel_mask::FIXED_WIDTH_NO_DICT}, true)) { return; } + if (!setupLocalPageInfo(s, pp, chunks, min_row, num_rows, + mask_filter{decode_kernel_mask::FIXED_WIDTH_NO_DICT}, + page_processing_stage::DECODE)) { + return; + } // must come after the kernel mask check [[maybe_unused]] null_count_back_copier _{s, t}; @@ -420,7 +424,10 @@ __global__ void __launch_bounds__(decode_block_size) gpuDecodePageDataFixedDict( if (!(BitAnd(pages[page_idx].kernel_mask, decode_kernel_mask::FIXED_WIDTH_DICT))) { return; } // TODO: abellina all_types_filter??? - if (!setupLocalPageInfo(s, pp, chunks, min_row, num_rows, mask_filter{decode_kernel_mask::FIXED_WIDTH_DICT}, true)) { return; } + if (!setupLocalPageInfo( + s, pp, chunks, min_row, num_rows, + mask_filter{decode_kernel_mask::FIXED_WIDTH_DICT}, + page_processing_stage::DECODE)) { return; } #ifdef ABDEBUG if (t == 0) { diff --git a/cpp/src/io/parquet/page_string_decode.cu b/cpp/src/io/parquet/page_string_decode.cu index 25b728632ed..a0ca043a15f 100644 --- a/cpp/src/io/parquet/page_string_decode.cu +++ b/cpp/src/io/parquet/page_string_decode.cu @@ -616,12 +616,13 @@ __global__ void __launch_bounds__(preprocess_block_size) gpuComputeStringPageBou decoders[level_type::NUM_LEVEL_TYPES] = {{def_runs}, {rep_runs}}; // setup page info + auto const mask = BitOr(decode_kernel_mask::STRING, decode_kernel_mask::DELTA_BYTE_ARRAY); if (!setupLocalPageInfo(s, pp, chunks, min_row, num_rows, - mask_filter{STRINGS_MASK}, + mask_filter{mask}, page_processing_stage::STRING_BOUNDS)) { return; }