Go to the documentation of this file.
6 #ifndef XGBOOST_COMMON_COMMON_H_
7 #define XGBOOST_COMMON_COMMON_H_
10 #include <xgboost/logging.h>
17 #include <type_traits>
23 #if defined(__CUDACC__)
24 #include <thrust/system/cuda/error.h>
25 #include <thrust/system_error.h>
27 #define WITH_CUDA() true
31 #define WITH_CUDA() false
33 #endif // defined(__CUDACC__)
36 #if defined(__CUDACC__)
40 #define safe_cuda(ans) ThrowOnCudaError((ans), __FILE__, __LINE__)
42 inline cudaError_t ThrowOnCudaError(cudaError_t code,
const char *file,
44 if (code != cudaSuccess) {
45 LOG(FATAL) << thrust::system_error(code, thrust::cuda_category(),
46 std::string{file} +
": " +
47 std::to_string(line)).what();
51 #endif // defined(__CUDACC__)
61 inline std::vector<std::string>
Split(
const std::string& s,
char delim) {
63 std::istringstream is(s);
64 std::vector<std::string> ret;
65 while (std::getline(is, item, delim)) {
79 std::ostringstream os;
84 template <
typename T1,
typename T2>
86 return static_cast<T1
>(std::ceil(
static_cast<double>(a) / b));
112 return i_ >= other.i_;
115 return i_ < other.i_;
123 i_{start}, step_{step} {}
140 return *begin_ == *other.begin_ && *end_ == *other.end_;
143 return !(*
this == other);
156 #ifndef XGBOOST_USE_CUDA
157 LOG(FATAL) <<
"XGBoost version not compiled with GPU support.";
158 #endif // XGBOOST_USE_CUDA
162 #ifndef XGBOOST_USE_ONEAPI
163 LOG(FATAL) <<
"XGBoost version not compiled with OneAPI support.";
164 #endif // XGBOOST_USE_ONEAPI
167 template <
typename Idx,
typename Container,
168 typename V =
typename Container::value_type,
169 typename Comp = std::less<V>>
170 std::vector<Idx>
ArgSort(Container
const &array, Comp comp = std::less<V>{}) {
171 std::vector<Idx> result(array.size());
172 std::iota(result.begin(), result.end(), 0);
173 auto op = [&array, comp](Idx
const &l, Idx
const &r) {
return comp(array[l], array[r]); };
179 #endif // XGBOOST_COMMON_COMMON_H_
XGBOOST_DEVICE Iterator begin() const
Definition: common.h:130
XGBOOST_DEVICE bool operator==(const Iterator &other) const
Definition: common.h:111
XGBOOST_DEVICE T Max(T a, T b)
Definition: common.h:72
XGBOOST_DEVICE bool operator==(const Range &other) const
Definition: common.h:139
XGBOOST_DEVICE Iterator(DifferenceType start)
Definition: common.h:121
XGBOOST_DEVICE DifferenceType operator*() const
Definition: common.h:100
#define XGBOOST_PARALLEL_STABLE_SORT(X, Y, Z)
Definition: base.h:69
defines configuration macros of xgboost.
XGBOOST_DEVICE Iterator operator++(int)
Definition: common.h:105
const XGBOOST_DEVICE Iterator & operator++()
Definition: common.h:101
XGBOOST_DEVICE Iterator end() const
Definition: common.h:131
int64_t DifferenceType
Definition: common.h:94
XGBOOST_DEVICE void Step(DifferenceType s)
Definition: common.h:118
void AssertGPUSupport()
Definition: common.h:155
std::vector< Idx > ArgSort(Container const &array, Comp comp=std::less< V >{})
Definition: common.h:170
XGBOOST_DEVICE Range(DifferenceType begin, DifferenceType end)
Definition: common.h:133
XGBOOST_DEVICE Range(DifferenceType begin, DifferenceType end, DifferenceType step)
Definition: common.h:135
XGBOOST_DEVICE T1 DivRoundUp(const T1 a, const T2 b)
Definition: common.h:85
std::string ToString(const T &data)
Definition: common.h:78
XGBOOST_DEVICE bool operator!=(const Range &other) const
Definition: common.h:142
XGBOOST_DEVICE Iterator(DifferenceType start, DifferenceType step)
Definition: common.h:122
XGBOOST_DEVICE bool operator!=(const Iterator &other) const
Definition: common.h:114
XGBOOST_DEVICE void Step(DifferenceType s)
Definition: common.h:146
void AssertOneAPISupport()
Definition: common.h:161
#define XGBOOST_DEVICE
Tag function as usable by device.
Definition: base.h:84
namespace of xgboost
Definition: base.h:110
std::vector< std::string > Split(const std::string &s, char delim)
Split a string by delimiter.
Definition: common.h:61