-
Notifications
You must be signed in to change notification settings - Fork 3.8k
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
WIP: [CUDA] New CUDA version #4528
Conversation
stop when no leaf to split
Does new version reuse current CUDA histograms building code? |
@StrikerRUS No, there's no dependence on the old CUDA histogram code. |
@shiyu1994 Maybe then we can name new version as |
@StrikerRUS I agree. Let's add the new CUDA version as |
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.
Just leaving a "request changes" review to be sure this isn't accidentally merged until we've agreed that enough time has passed since 3.3.0.
Relevant comment: #4633 (comment)
How can I build this branch? |
@GeorgePearse Thanks for using LightGBM. To build this branch, you may clone from https://github.com/shiyu1994/LightGBM/tree/cuda-new-refactor-2 and follow the build instruction of LightGBM CUDA version here. https://lightgbm.readthedocs.io/en/latest/Installation-Guide.html#build-cuda-version-experimental |
@shiyu1994 Thanks! Just need to try to find the fastest training model possible. Stability not a worry. Is it useful for me to add any bugs I encounter here? Or is there good enough test coverage that I'll just be restating what's already known. |
@GeorgePearse Yes, if you find any bug you may post it here. There are some features that CUDA version of this branch does not support, including training with categorical features. Other additional features during training, like monotonic constraints, extremely random trees are not supported. I'm now working with #4630, which only dispatches the tree learner part to CUDA. That branch is relatively more complete. I'll finish #4630 first, and then merge additional contents in this PR. Thanks for your support. |
@shiyu1994 is it still your plan to merge |
@jameslamb I believe we should close this. I'll open additional PRs for the rest CUDA work. Thanks! |
This pull request has been automatically locked since there has not been any recent activity since it was closed. To start a new related discussion, open a new issue at https://github.com/microsoft/LightGBM/issues including a reference to this. |
Current GPU and CUDA versions of LightGBM only calculate histograms on GPU. This is a new CUDA version for LightGBM which runs the whole training process on GPU.