Rory Mitchell
d686bf52a6
Reduce time for some multi-gpu tests ( #8288 )
...
* Faster dask tests
* Reuse AllReducer objects in tests.
* Faster boost from prediction tests.
* Use rmm dask fixture.
* Speed up dask demo.
* mypy
* Format with black.
* mypy
* Clang-tidy
Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
2022-10-04 02:49:33 -08:00
Jiaming Yuan
8fccc3c4ad
[dask] Fix potential error in demo. ( #8079 )
...
* Use dask_cudf instead.
2022-07-15 18:42:29 +08:00
Jiaming Yuan
ec56d5869b
[doc] Include dask examples into doc. ( #7530 )
2022-01-05 03:27:22 +08:00
Jiaming Yuan
78d72ef936
Add DaskDeviceQuantileDMatrix demo. ( #6156 )
2020-09-24 14:08:28 +08:00
Jiaming Yuan
761a5dbdfc
[dask] Honor nthreads from dask worker. ( #5414 )
2020-03-16 04:51:24 +08:00
Jiaming Yuan
7e72a12871
Don't set_params at the end of set_state. ( #4947 )
...
* Don't set_params at the end of set_state.
* Also fix another issue found in dask prediction.
* Add note about prediction.
Don't support other prediction modes at the moment.
2019-10-15 10:08:26 -04:00
Jiaming Yuan
2ebdec8aa6
Fix dask prediction. ( #4941 )
...
* Fix dask prediction.
* Add better error messages for wrong partition.
2019-10-14 23:19:34 -04:00
Jiaming Yuan
7e24a8d245
Improve doc and demo for dask. ( #4907 )
...
* Add a readme with link to doc.
* Add more comments in the demonstrations code.
* Workaround https://github.com/dask/distributed/issues/3081 .
2019-09-30 23:59:37 -04:00
Jiaming Yuan
b8433c455a
Rewrite Dask interface. ( #4819 )
2019-09-25 01:30:14 -04:00