-
Notifications
You must be signed in to change notification settings - Fork 915
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
Enable segmented_gather in Java package #10669
Enable segmented_gather in Java package #10669
Conversation
Signed-off-by: sperlingxx <[email protected]>
Codecov Report
@@ Coverage Diff @@
## branch-22.06 #10669 +/- ##
================================================
+ Coverage 86.34% 86.41% +0.06%
================================================
Files 142 142
Lines 22356 22334 -22
================================================
- Hits 19304 19300 -4
+ Misses 3052 3034 -18
Continue to review full report at Codecov.
|
try { | ||
cudf::jni::auto_set_device(env); | ||
auto src_col = reinterpret_cast<cudf::lists_column_view *>(source_column); | ||
auto gather_map = reinterpret_cast<cudf::lists_column_view *>(gather_map_list); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please cast these to a cudf::column_view
and then create a cudf::lists_column_view
from them. Otherwise there is no checking that the columns passed in are really lists. If they are not really bad things could happen.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done.
@gpucibot merge |
Current PR is to enable cuDF API
segmented_gather
in Java package.segmented_gather
is essential to implement spark array functions likearrays_zip
(NVIDIA/spark-rapids#5229).