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

spillable cache for GpuCartesianRDD #1878

Merged
merged 5 commits into from
Mar 16, 2021

Conversation

sperlingxx
Copy link
Collaborator

Use SpillableColumnarBatch to cache stream-side data in case of re-computation caused by nested-loop. Current PR is the substitution of #1784, which based on branch-0.4.

Signed-off-by: sperlingxx lovedreamf@gmail.com

Signed-off-by: sperlingxx <lovedreamf@gmail.com>
@sperlingxx
Copy link
Collaborator Author

build

@jlowe jlowe added the SQL part of the SQL/Dataframe plugin label Mar 5, 2021
@sperlingxx
Copy link
Collaborator Author

build

Signed-off-by: sperlingxx <lovedreamf@gmail.com>
@sperlingxx
Copy link
Collaborator Author

build

// happen if there exists specific plans like `LimitExec`.
context.addTaskCompletionListener[Unit]((_: TaskContext) => {
spillBatchBuffer.safeClose()
if (buildTableOnFlight.isDefined) buildTableOnFlight.get.close()
Copy link
Member

Choose a reason for hiding this comment

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

Suggested change
if (buildTableOnFlight.isDefined) buildTableOnFlight.get.close()
buildTableOnFlight.foreach(_.close())

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

Change applied, Thx!

@sperlingxx
Copy link
Collaborator Author

build

1 similar comment
@sperlingxx
Copy link
Collaborator Author

build

@sperlingxx
Copy link
Collaborator Author

build

@revans2 revans2 merged commit a4a72f3 into NVIDIA:branch-0.5 Mar 16, 2021
@sperlingxx sperlingxx deleted the spill_cart_rdd2 branch April 8, 2021 03:04
nartal1 pushed a commit to nartal1/spark-rapids that referenced this pull request Jun 9, 2021
Signed-off-by: sperlingxx <lovedreamf@gmail.com>
nartal1 pushed a commit to nartal1/spark-rapids that referenced this pull request Jun 9, 2021
Signed-off-by: sperlingxx <lovedreamf@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
SQL part of the SQL/Dataframe plugin
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants