Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix ColumnarToRowIterator handling of empty batches #2318

Merged
merged 1 commit into from
May 1, 2021

Conversation

jlowe
Copy link
Member

@jlowe jlowe commented May 1, 2021

Fixes #2317

This updates loadNextBatch to pull from the input batch iterator in a loop as long as there are empty batches being received.

Signed-off-by: Jason Lowe <jlowe@nvidia.com>
@jlowe jlowe added the bug Something isn't working label May 1, 2021
@jlowe jlowe self-assigned this May 1, 2021
@jlowe
Copy link
Member Author

jlowe commented May 1, 2021

build

// but it is more efficient.
numOutputRows += cb.numRows()
}
return true
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: I think it is more scala like to have it be functional.

if (devCb.numRows > 0) {
  ...
  true
} else {
  false
}

@revans2 revans2 merged commit 518a5b6 into NVIDIA:branch-0.5 May 1, 2021
nartal1 pushed a commit to nartal1/spark-rapids that referenced this pull request Jun 9, 2021
Signed-off-by: Jason Lowe <jlowe@nvidia.com>
nartal1 pushed a commit to nartal1/spark-rapids that referenced this pull request Jun 9, 2021
Signed-off-by: Jason Lowe <jlowe@nvidia.com>
@jlowe jlowe deleted the fix-columnartorow-iterator branch September 10, 2021 15:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[BUG] GpuColumnarToRowIterator can stop after receiving an empty batch
2 participants