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

Fix _ray_get_act_cpus and set nthread in param passed to DMatrix creation #274

Merged
merged 3 commits into from
Mar 6, 2023

Conversation

kira-lin
Copy link
Contributor

@kira-lin kira-lin commented Mar 3, 2023

This pr closes #266

_ray_get_act_cpus did not get the correct number set for CPU in actors, because when placement group is used, the resource name changes. In addition, set this number as nthread in param which is used to create DMatrix. @Yard1

Copy link
Member

@Yard1 Yard1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks, this is great! Could we get a quick unit test for this? just checking if _ray_get_actor_cpus() returns the expected value.

xgboost_ray/main.py Show resolved Hide resolved
@Yard1 Yard1 merged commit 44e0f01 into ray-project:master Mar 6, 2023
@Yard1
Copy link
Member

Yard1 commented Mar 6, 2023

Thanks!

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.

Long single-core period during training hurts performance
2 participants