xgboost

Hendrik/xgboost

Fork 0

Commit Graph

Author	SHA1	Message	Date
Jiaming Yuan	c589eff941	De-duplicate GPU parameters. (#4454 ) * Only define `gpu_id` and `n_gpus` in `LearnerTrainParam` * Pass LearnerTrainParam through XGBoost vid factory method. * Disable all GPU usage when GPU related parameters are not specified (fixes XGBoost choosing GPU over aggressively). * Test learner train param io. * Fix gpu pickling.	2019-05-29 11:55:57 +08:00
Jiaming Yuan	f1275f52c1	Fix specifying gpu_id, add tests. (#3851 ) * Rewrite gpu_id related code. * Remove normalised/unnormalised operatios. * Address difference between `Index' and `Device ID'. * Modify doc for `gpu_id'. * Better LOG for GPUSet. * Check specified n_gpus. * Remove inappropriate `device_idx' term. * Clarify GpuIdType and size_t.	2018-11-06 18:17:53 +13:00
trivialfis	9119f9e369	Fix gpu devices. (#3693 ) * Fix gpu_set normalized and unnormalized. * Fix DeviceSpan.	2018-09-19 17:39:42 +12:00

Author

SHA1

Message

Date

Jiaming Yuan

c589eff941

De-duplicate GPU parameters. (#4454 )

* Only define `gpu_id` and `n_gpus` in `LearnerTrainParam`
* Pass LearnerTrainParam through XGBoost vid factory method.
* Disable all GPU usage when GPU related parameters are not specified (fixes XGBoost choosing GPU over aggressively).
* Test learner train param io.
* Fix gpu pickling.

2019-05-29 11:55:57 +08:00

Jiaming Yuan

f1275f52c1

Fix specifying gpu_id, add tests. (#3851 )

* Rewrite gpu_id related code.

* Remove normalised/unnormalised operatios.
* Address difference between `Index' and `Device ID'.
* Modify doc for `gpu_id'.
* Better LOG for GPUSet.
* Check specified n_gpus.
* Remove inappropriate `device_idx' term.
* Clarify GpuIdType and size_t.

2018-11-06 18:17:53 +13:00

trivialfis

9119f9e369

Fix gpu devices. (#3693 )

* Fix gpu_set normalized and unnormalized.
* Fix DeviceSpan.

2018-09-19 17:39:42 +12:00

3 Commits