-
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
[JNI] Remove child fom newCudaAsyncMemoryResource #12681
[JNI] Remove child fom newCudaAsyncMemoryResource #12681
Conversation
Codecov Report
Additional details and impacted files@@ Coverage Diff @@
## branch-23.04 #12681 +/- ##
===============================================
Coverage ? 85.81%
===============================================
Files ? 158
Lines ? 25153
Branches ? 0
===============================================
Hits ? 21586
Misses ? 3567
Partials ? 0 Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. ☔ View full report at Codecov. |
Dask/python tests failing it looks like. |
…da_async_memory_resource
Upmerged to see if that helps the CI pass, since I see some related python changes went in. |
Also: |
/merge |
Fixes a bug introduced here #12632 where the C++ version of the jni call
Rmm.newCudaAsyncMemoryResource
was taking an extra argument (child
), causing thethreshold
to be set to a random value since the calling code defined the native method has taking 2 longs, not 3.The above can affect performance if the threshold is set to 0 for example. The async pool will reduce its footprint by releasing memory, which means we need to go back and reallocate. This is how I found it.
I am starting the build process for this, so it's going to take me a bit to test.