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

Add UDT support to ParquetCachedBatchSerializer (CPU) #4955

Merged
merged 6 commits into from
Mar 21, 2022

Conversation

razajafri
Copy link
Collaborator

@razajafri razajafri commented Mar 15, 2022

This PR adds UDT support for the case when PCBS fallsback to CPU and make sure we don't put stuff on the GPU if the output of InMemoryRelation is not supported.

fixes #4806

Signed-off-by: Raza Jafri <rjafri@nvidia.com>
Signed-off-by: Raza Jafri <rjafri@nvidia.com>
Signed-off-by: Raza Jafri <rjafri@nvidia.com>
@razajafri
Copy link
Collaborator Author

build

@razajafri razajafri self-assigned this Mar 15, 2022
@razajafri
Copy link
Collaborator Author

build

Signed-off-by: Raza Jafri <rjafri@nvidia.com>
@razajafri razajafri requested a review from revans2 March 15, 2022 18:25
@razajafri
Copy link
Collaborator Author

build

@sameerz sameerz added the bug Something isn't working label Mar 16, 2022
@sameerz sameerz added this to the Feb 28 - Mar 18 milestone Mar 16, 2022
Signed-off-by: Raza Jafri <rjafri@nvidia.com>
@razajafri
Copy link
Collaborator Author

build

@razajafri
Copy link
Collaborator Author

@revans2 have I addressed your concerns?

Signed-off-by: Raza Jafri <rjafri@nvidia.com>
@razajafri
Copy link
Collaborator Author

build

@razajafri razajafri merged commit c3cf357 into NVIDIA:branch-22.04 Mar 21, 2022
@razajafri razajafri deleted the SR-4806-PCBS-udt branch March 21, 2022 22:33
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] When running xgboost training, if PCBS is enabled, it fails with java.lang.AssertionError
3 participants