-
Notifications
You must be signed in to change notification settings - Fork 236
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 in new HostAlloc API #9170
Add in new HostAlloc API #9170
Conversation
Signed-off-by: Robert (Bobby) Evans <[email protected]>
sql-plugin/src/main/scala/com/nvidia/spark/rapids/HostAlloc.scala
Outdated
Show resolved
Hide resolved
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.
I've done one pass through the API. Overall it's looking good, I mostly had high level questions. Will wait for feedback and move onto tests sometime today.
|
||
import org.apache.spark.TaskContext | ||
|
||
private class HostAlloc(nonPinnedLimit: Long) { |
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.
nit, should we used "pageable" instead of "non-pinned"?
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.
If you really want me to I will. Is it unclear the type of memory being allocated?
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.
It's a nit, not a must fix. It is clear to me that non-pinned is pageable.
|
||
import org.apache.spark.TaskContext | ||
|
||
private class HostAlloc(nonPinnedLimit: Long) { |
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.
suggest we call this HostAllocator
.
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.
If you really want me to I will. Is it unclear what this does?
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.
the main nit I had was that HostAlloc sounded more like a product of an allocator than the allocator itself, but it's just a nit. Not a must fix.
sql-plugin/src/main/scala/com/nvidia/spark/rapids/HostAlloc.scala
Outdated
Show resolved
Hide resolved
sql-plugin/src/main/scala/com/nvidia/spark/rapids/HostAlloc.scala
Outdated
Show resolved
Hide resolved
sql-plugin/src/main/scala/com/nvidia/spark/rapids/HostAlloc.scala
Outdated
Show resolved
Hide resolved
build |
sql-plugin/src/main/scala/com/nvidia/spark/rapids/HostAlloc.scala
Outdated
Show resolved
Hide resolved
build |
This fixes #8879
The dependency for this was merged into CUDF today, but it may be a little while before this can be merged/tested.