Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
5,869 Commits 32 Branches 59 Tags
Commit Graph

2 Commits

Author SHA1 Message Date
Philip Hyunsu Cho
366f3cb9d8
Add use_rmm flag to global configuration (#6656) 
* Ensure RMM is 0.18 or later

* Add use_rmm flag to global configuration

* Modify XGBCachingDeviceAllocatorImpl to skip CUB when use_rmm=True

* Update the demo

* [CI] Pin NumPy to 1.19.4, since NumPy 1.19.5 doesn't work with latest Shap
 2021-03-09 14:53:05 -08:00
Philip Hyunsu Cho
fb56da5e8b
Add global configuration (#6414) 
* Add management functions for global configuration: XGBSetGlobalConfig(), XGBGetGlobalConfig().
* Add Python interface: set_config(), get_config(), and config_context().
* Add unit tests for Python
* Add R interface: xgb.set.config(), xgb.get.config()
* Add unit tests for R

Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
 2020-12-03 00:05:18 -08:00