Fix _ray_get_act_cpus and set nthread in param passed to DMatrix creation

ray-project / xgboost_ray

Distributed XGBoost on Ray

Apache License 2.0

133 stars 34 forks source link

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

Closed kira-lin closed 1 year ago

kira-lin commented 1 year ago

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

Yard1 commented 1 year ago

Thanks!