-
Notifications
You must be signed in to change notification settings - Fork 222
init parallel trees method #3139
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
init parallel trees method #3139
Conversation
/intelci: run |
/intelci: run |
/intelci: run |
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.
ABI checks looks as false positive, as adding virtual methods into a virtual class shouldn't be an issue. But maybe I am getting something wrong.
Please take a look at the other comments.
/intelci: run |
/intelci: run |
/intelci: run |
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.
Let's wait for the CI and LGTM.
Description
Summary
This PR introduces a new method
local_trees
for the Random Forest implementation. The method builds local trees independently on each GPU node, enabling efficient parallelization and improved scaling.Key Changes
local_trees
method for distributed tree construction across GPUS.serialize local models -> allgather local models -> deserialize local models -> copy local models in the global model).
PR completeness and readability
Testing
Performance