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
..
2021-02-09 08:12:58 +08:00
2021-03-04 14:37:20 +08:00
2021-02-09 23:30:19 +08:00
2021-02-09 23:30:19 +08:00
2020-08-12 01:26:02 -07:00