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

Add an explicit option to use RMM #6297

Closed
hcho3 opened this issue Oct 27, 2020 · 0 comments · Fixed by #6656
Closed

Add an explicit option to use RMM #6297

hcho3 opened this issue Oct 27, 2020 · 0 comments · Fixed by #6656

Comments

@hcho3
Copy link
Collaborator

hcho3 commented Oct 27, 2020

Moved from rapidsai/rmm#583.

We should ask the user to explicitly declare that they are using RMM allocator. The desired behavior is as follows:

  • If use_rmm is set to True: XGBoost will use RMM allocator for every allocation, trusting the users to set a suitable device resource for good performance.
  • If use_rmm is set to False: XGBoost will use CUB allocator as well as plain cudaMalloc(). We trust the users to remember setting use_rmm=True when they set up an RMM memory pool.
  • use_rmm should be set to False by default.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants