xgboost/test_aft_obj.cu at 15421e40d9c38c0083c02d0aa426f5957c27c30d - xgboost - Gitea: Git with a cup of tea

Hendrik/xgboost

Philip Hyunsu Cho 71b0528a2f

GPU implementation of AFT survival objective and metric (#5714 )

* Add interval accuracy

* De-virtualize AFT functions

* Lint

* Refactor AFT metric using GPU-CPU reducer

* Fix R build

* Fix build on Windows

* Fix copyright header

* Clang-tidy

* Fix crashing demo

* Fix typos in comment; explain GPU ID

* Remove unnecessary #include

* Add C++ test for interval accuracy

* Fix a bug in accuracy metric: use log pred

* Refactor AFT objective using GPU-CPU Transform

* Lint

* Fix lint

* Use Ninja to speed up build

* Use time, not /usr/bin/time

* Add cpu_build worker class, with concurrency = 1

* Use concurrency = 1 only for CUDA build

* concurrency = 1 for clang-tidy

* Address reviewer's feedback

* Update link to AFT paper

2020-07-17 01:18:13 -07:00

7 lines

113 B

Plaintext

Raw Blame History

 /*!
  * Copyright 2020 XGBoost contributors
  */
 // Dummy file to keep the CUDA tests.
 #include "test_aft_obj.cc"