enable ROCm on latest XGBoost
This commit is contained in:
@@ -6,7 +6,6 @@
|
||||
#ifndef XGBOOST_COMMON_COMMON_H_
|
||||
#define XGBOOST_COMMON_COMMON_H_
|
||||
|
||||
#include <algorithm> // for max
|
||||
#include <array> // for array
|
||||
#include <cmath> // for ceil
|
||||
#include <cstddef> // for size_t
|
||||
@@ -203,7 +202,7 @@ inline void SetDevice(std::int32_t device) {
|
||||
#endif
|
||||
|
||||
/**
|
||||
* Last index of a group in a CSR style of index pointer.
|
||||
* @brief Last index of a group in a CSR style of index pointer.
|
||||
*/
|
||||
template <typename Indexable>
|
||||
XGBOOST_DEVICE size_t LastOf(size_t group, Indexable const &indptr) {
|
||||
|
||||
Reference in New Issue
Block a user