Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow
Hello, I noticed that the "GPU Support" section of the XGBoost documentation has recently been updated to exclude the table of objective functions with GPU implementations.
From the previous documentation, it seems the only objective function without GPU implementation is survival:cox. I checked the current implementation of CoxRegression in src/objective/regression_obj.cu, and it seems its GetGradient function still doesn't make use of GPU parallelism.
As someone who would like to work on implementing GPU support for survival:cox, I wonder whether the documentation change reflects a shift of development focus, or it is still encouraged to try to implement GPU support for survival:cox?
Hello, I noticed that the "GPU Support" section of the XGBoost documentation has recently been updated to exclude the table of objective functions with GPU implementations.
From the previous documentation, it seems the only objective function without GPU implementation is
survival:cox
. I checked the current implementation ofCoxRegression
insrc/objective/regression_obj.cu
, and it seems itsGetGradient
function still doesn't make use of GPU parallelism.As someone who would like to work on implementing GPU support for
survival:cox
, I wonder whether the documentation change reflects a shift of development focus, or it is still encouraged to try to implement GPU support forsurvival:cox
?Thank you!
Figure: previous XGBoost documentation