Skip to content

Commit

Permalink
Fix a minor potential issue when rebatching for GpuArrowEvalPythonExec (
Browse files Browse the repository at this point in the history
NVIDIA#5995)

There is a chance that the row number of the combined batches is
just a multiple of batch size, which results in returning this huge
ColumnBatch directly.

Signed-off-by: Bobby Wang <wbo4958@gmail.com>
  • Loading branch information
wbo4958 authored Jul 14, 2022
1 parent af687db commit 4a313e7
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ class RebatchingRoundoffIterator(

val rc: Long = combined.numRows()

if (rc % targetRoundoff == 0 || rc < targetRoundoff) {
if (rc <= targetRoundoff) {
return combined
}

Expand Down

0 comments on commit 4a313e7

Please sign in to comment.