From cca9be6252ff4bd808afe1ef0a6b9d19895a3a77 Mon Sep 17 00:00:00 2001 From: seidl Date: Wed, 3 Jan 2024 15:39:08 -0800 Subject: [PATCH] do not exit loop prematurely on error --- cpp/src/io/parquet/page_decode.cuh | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/cpp/src/io/parquet/page_decode.cuh b/cpp/src/io/parquet/page_decode.cuh index a378ea33a0c..8f256cd1f97 100644 --- a/cpp/src/io/parquet/page_decode.cuh +++ b/cpp/src/io/parquet/page_decode.cuh @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2023, NVIDIA CORPORATION. + * Copyright (c) 2018-2024, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -506,14 +506,9 @@ __device__ void gpuDecodeStream( cur++; } } - if (cur > end) { - s->set_error_code(decode_error::LEVEL_STREAM_OVERRUN); - break; - } - if (level_run <= 1) { - s->set_error_code(decode_error::INVALID_LEVEL_RUN); - break; - } + // If there are errors, set the error code and continue. The loop will be exited below. + if (cur > end) { s->set_error_code(decode_error::LEVEL_STREAM_OVERRUN); } + if (level_run <= 1) { s->set_error_code(decode_error::INVALID_LEVEL_RUN); } sym_len = (int32_t)(cur - cur_def); __threadfence_block(); }