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

[JNI] Pass this ColumnVector to the onClosed event handler #13386

Merged

Conversation

abellina
Copy link
Contributor

Putting this up as an optimization for ColumnVector.EventHandler.

As it stands today, code in the spark-rapids plugin that wants to use this is having to java objects that encapsulate state about the columns around reference counting and what represents the bag of ColumnVectors that are spillable at any given time. If we pass the ColumnVector instance in the event handler, one of these objects can be removed, and we can simplify the implementation in the plugin. I am putting this up as draft while my local tests pass, but I think it should be fairly straightforward.

@github-actions github-actions bot added the Java Affects Java cuDF API. label May 18, 2023
@abellina abellina added improvement Improvement / enhancement to an existing function Spark Functionality that helps Spark RAPIDS non-breaking Non-breaking change labels May 18, 2023
@abellina
Copy link
Contributor Author

Marked as non-breaking because code that would use it is not checked in to spark-rapids yet.

@abellina abellina changed the title Pass this ColumnVector to the onClosed event handler [JNI] Pass this ColumnVector to the onClosed event handler May 18, 2023
@abellina abellina marked this pull request as ready for review May 19, 2023 20:59
@abellina abellina requested a review from a team as a code owner May 19, 2023 20:59
@abellina abellina self-assigned this May 19, 2023
@abellina
Copy link
Contributor Author

/merge

@rapids-bot rapids-bot bot merged commit 209ab6e into rapidsai:branch-23.06 May 19, 2023
@abellina abellina deleted the pass_column_vector_ref_to_onClosed branch May 19, 2023 21:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
improvement Improvement / enhancement to an existing function Java Affects Java cuDF API. non-breaking Non-breaking change Spark Functionality that helps Spark RAPIDS
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants