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

Change order of initialization so pinned pool is available for spill framework buffers #11889

Open
wants to merge 1 commit into
base: branch-25.02
Choose a base branch
from

Conversation

abellina
Copy link
Collaborator

@abellina abellina commented Dec 18, 2024

While collecting traces for #11888, I realized that the bounce buffers I am using are pageable, which is pretty bad. If I change the order of initialization, the pinned pool can be initialized before the spill framework, so we can then use pinned memory for the d2h and h2d bounced copies.

Renamed init functions slightly, as their names are not matching their function currently.

@abellina
Copy link
Collaborator Author

build

@abellina
Copy link
Collaborator Author

CI failures due to tests initializing too many times. Before my change we had all of this code under a "make sure Rmm is not initialized already" block.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant