Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix device dispatch for linear updater. (#9507)

Browse Source
This commit is contained in:
Jiaming Yuan
2023-08-23 00:17:35 +08:00
committed by GitHub
parent 302bbdc958
commit 3c09399f29
6 changed files with 80 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
doc/parameter.rst
View File

@@ -329,7 +329,7 @@ Parameters for Linear Booster (``booster=gblinear``)
- Choice of algorithm to fit linear model
- ``shotgun``: Parallel coordinate descent algorithm based on shotgun algorithm. Uses 'hogwild' parallelism and therefore produces a nondeterministic solution on each run.
- ``coord_descent``: Ordinary coordinate descent algorithm. Also multithreaded but still produces a deterministic solution.
- ``coord_descent``: Ordinary coordinate descent algorithm. Also multithreaded but still produces a deterministic solution. When the ``device`` parameter is set to ``cuda`` or ``gpu``, a GPU variant would be used.
* ``feature_selector`` [default= ``cyclic``]