-
-
Notifications
You must be signed in to change notification settings - Fork 8.7k
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
Support column split in gpu hist updater #9384
Conversation
@trivialfis I'm working on enabling column-split with the gpu hist tree method. Only added some tests here, but they seem to hang on some cuda calls. Any idea on how to debug this? Do you see anything obviously wrong? |
@rongou GPU doesn't support multi-target yet. |
@trivialfis I think this is finally working. Please take a look. Thanks! |
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.
Looks good to me!
Would you like to produce a document that briefly introduces how everything works with column split in the future?
Yes probably should create a tutorial. |
For some reason, this PR has optimized the GPU hist with the Higgs dataset for about 20 percent improvement, tested with both RTX8000 and 4070 ti super. |
NVM, git bisect is incorrect. |
No description provided.