29 #ifndef XGBOOST_SPAN_H_
30 #define XGBOOST_SPAN_H_
33 #include <xgboost/logging.h>
38 #include <type_traits>
41 #if defined(__CUDACC__)
42 #include <cuda_runtime.h>
62 #if defined(_MSC_VER) && _MSC_VER < 1910
64 #define __span_noexcept
66 #pragma push_macro("constexpr")
71 #define __span_noexcept noexcept
78 #if defined(__CUDA_ARCH__)
84 #define CUDA_KERNEL_CHECK(cond) \
86 if (XGBOOST_EXPECT(!(cond), false)) { \
93 #define __ASSERT_STR_HELPER(x) #x
95 #define CUDA_KERNEL_CHECK(cond) \
96 (XGBOOST_EXPECT((cond), true) \
97 ? static_cast<void>(0) \
98 : __assert_fail(__ASSERT_STR_HELPER((cond)), __FILE__, __LINE__, __PRETTY_FUNCTION__))
102 #define KERNEL_CHECK CUDA_KERNEL_CHECK
104 #define SPAN_CHECK KERNEL_CHECK
108 #if defined(XGBOOST_STRICT_R_MODE) && XGBOOST_STRICT_R_MODE == 1
110 #define KERNEL_CHECK(cond)
112 #define SPAN_CHECK(cond) KERNEL_CHECK(cond)
116 #define KERNEL_CHECK(cond) (XGBOOST_EXPECT((cond), true) ? static_cast<void>(0) : std::terminate())
118 #define SPAN_CHECK(cond) KERNEL_CHECK(cond)
124 #define SPAN_LT(lhs, rhs) SPAN_CHECK((lhs) < (rhs))
134 std::is_same<std::ptrdiff_t, std::int64_t>::value,
138 #if defined(_MSC_VER) && _MSC_VER < 1910
139 constexpr
const std::size_t
145 enum class byte : unsigned char {};
147 template <
class ElementType, std::
size_t Extent>
152 template <
typename SpanType,
bool IsConst>
154 using ElementType =
typename SpanType::element_type;
162 IsConst,
const ElementType, ElementType>::type&;
163 using pointer =
typename std::add_pointer<reference>::type;
168 const SpanType* _span,
173 template <bool B, typename std::enable_if<!B && IsConst>::type* =
nullptr>
243 return _lhs.span_ == _rhs.span_ && _lhs.index_ == _rhs.index_;
248 return !(_lhs == _rhs);
253 return _lhs.index_ < _rhs.index_;
258 return !(_rhs < _lhs);
268 return !(_rhs > _lhs);
273 typename SpanType::index_type
index_ { 0 };
287 template <std::
size_t Extent, std::
size_t Offset, std::
size_t Count>
289 std::size_t, Count != dynamic_extent ?
290 Count : (Extent != dynamic_extent ? Extent - Offset : Extent)> {};
296 template <
typename T, std::
size_t Extent>
299 Extent == dynamic_extent ?
300 Extent : sizeof(T) * Extent> {};
302 template <std::
size_t From, std::
size_t To>
304 bool, From == To || From == dynamic_extent || To == dynamic_extent> {};
306 template <
class From,
class To>
308 bool, std::is_convertible<From(*)[], To(*)[]>::value> {};
313 template <
class T, std::
size_t Extent>
320 template <
typename T>
327 template <
typename T>
334 template <
class InputIt1,
class InputIt2,
338 InputIt2 first2, InputIt2 last2) {
340 for (; first1 != last1 && first2 != last2; ++first1, ++first2) {
341 if (comp(*first1, *first2)) {
344 if (comp(*first2, *first1)) {
348 return first1 == last1 && first2 != last2;
421 template <
typename T,
441 size_(_count), data_(_ptr) {
447 size_(_last - _first), data_(_first) {
451 template <std::
size_t N>
455 template <
class Container,
456 class =
typename std::enable_if<
457 !std::is_const<element_type>::value &&
459 std::is_convertible<typename Container::pointer, pointer>::value &&
460 std::is_convertible<
typename Container::pointer,
461 decltype(std::declval<Container>().
data())>::value>::type>
463 size_(_cont.
size()), data_(_cont.
data()) {
467 template <
class Container,
468 class =
typename std::enable_if<
469 std::is_const<element_type>::value &&
471 std::is_convertible<typename Container::pointer, pointer>::value &&
472 std::is_convertible<
typename Container::pointer,
473 decltype(std::declval<Container>().
data())>::value>::type>
474 Span(
const Container& _cont) : size_(_cont.
size()),
475 data_(_cont.
data()) {
479 template <
class U, std::size_t OtherExtent,
480 class =
typename std::enable_if<
490 size_ = _other.size();
491 data_ = _other.data();
502 return {
this,
size()};
510 return {
this,
size()};
536 return (*
this)[
size() - 1];
557 return size() *
sizeof(T);
565 template <std::
size_t Count>
568 return {
data(), Count};
572 std::size_t _count)
const {
574 return {
data(), _count};
577 template <std::
size_t Count>
580 return {
data() +
size() - Count, Count};
584 std::size_t _count)
const {
593 template <std::size_t
Offset,
597 detail::ExtentValue<Extent,
Offset, Count>::value> {
607 (_offset <=
size()) : (_offset + _count <=
size()));
608 return {
data() + _offset, _count ==
617 template <
class T, std::
size_t X,
class U, std::
size_t Y>
624 if (*l_beg != *r_beg) {
631 template <
class T, std::
size_t X,
class U, std::
size_t Y>
636 template <
class T, std::
size_t X,
class U, std::
size_t Y>
642 template <
class T, std::
size_t X,
class U, std::
size_t Y>
647 template <
class T, std::
size_t X,
class U, std::
size_t Y>
655 template <
class T, std::
size_t X,
class U, std::
size_t Y>
660 template <
class T, std::
size_t E>
663 return {
reinterpret_cast<const byte*
>(s.data()), s.
size_bytes()};
666 template <
class T, std::
size_t E>
669 return {
reinterpret_cast<byte*
>(s.data()), s.
size_bytes()};
674 #if defined(_MSC_VER) &&_MSC_VER < 1910
676 #pragma pop_macro("constexpr")
677 #undef __span_noexcept
defines configuration macros of xgboost.
#define XGBOOST_DEVICE
Tag function as usable by device.
Definition: base.h:84
span class implementation, based on ISO++20 span<T>. The interface should be the same.
Definition: span.h:423
T & reference
Definition: span.h:430
XGBOOST_DEVICE Span(pointer _first, pointer _last)
Definition: span.h:446
T * pointer
Definition: span.h:429
detail::ptrdiff_t difference_type
Definition: span.h:428
XGBOOST_DEVICE reference operator()(index_type _idx) const
Definition: span.h:544
XGBOOST_DEVICE reference front() const
Definition: span.h:531
constexpr XGBOOST_DEVICE const_iterator cbegin() const __span_noexcept
Definition: span.h:505
constexpr XGBOOST_DEVICE iterator begin() const __span_noexcept
Definition: span.h:497
constexpr XGBOOST_DEVICE pointer data() const __span_noexcept
Definition: span.h:548
std::size_t index_type
Definition: span.h:427
XGBOOST_DEVICE reference back() const
Definition: span.h:535
XGBOOST_DEVICE auto subspan() const -> Span< element_type, detail::ExtentValue< Extent, Offset, Count >::value >
Definition: span.h:595
constexpr XGBOOST_DEVICE index_type size() const __span_noexcept
Definition: span.h:553
XGBOOST_DEVICE Span< element_type, Count > last() const
Definition: span.h:578
Span(Container &_cont)
Definition: span.h:462
XGBOOST_DEVICE Span< element_type, Count > first() const
Definition: span.h:566
constexpr XGBOOST_DEVICE Span(const Span &_other) __span_noexcept
Definition: span.h:486
constexpr XGBOOST_DEVICE bool empty() const __span_noexcept
Definition: span.h:560
constexpr XGBOOST_DEVICE index_type size_bytes() const __span_noexcept
Definition: span.h:556
constexpr XGBOOST_DEVICE Span(element_type(&arr)[N]) __span_noexcept
Definition: span.h:452
constexpr reverse_iterator rbegin() const __span_noexcept
Definition: span.h:513
T element_type
Definition: span.h:425
const std::reverse_iterator< const_iterator > const_reverse_iterator
Definition: span.h:435
XGBOOST_DEVICE Span< element_type, dynamic_extent > first(std::size_t _count) const
Definition: span.h:571
constexpr XGBOOST_DEVICE const_reverse_iterator crbegin() const __span_noexcept
Definition: span.h:521
constexpr XGBOOST_DEVICE const_iterator cend() const __span_noexcept
Definition: span.h:509
XGBOOST_DEVICE ~Span() __span_noexcept
Definition: span.h:495
constexpr XGBOOST_DEVICE const_reverse_iterator crend() const __span_noexcept
Definition: span.h:525
constexpr Span() __span_noexcept=default
constexpr XGBOOST_DEVICE iterator end() const __span_noexcept
Definition: span.h:501
XGBOOST_DEVICE reference operator[](index_type _idx) const
Definition: span.h:539
constexpr reverse_iterator rend() const __span_noexcept
Definition: span.h:517
XGBOOST_DEVICE Span & operator=(const Span &_other) __span_noexcept
Definition: span.h:489
std::reverse_iterator< iterator > reverse_iterator
Definition: span.h:434
XGBOOST_DEVICE Span< element_type, dynamic_extent > subspan(index_type _offset, index_type _count=dynamic_extent) const
Definition: span.h:603
Span(const Container &_cont)
Definition: span.h:474
constexpr XGBOOST_DEVICE Span(const Span< U, OtherExtent > &_other) __span_noexcept
Definition: span.h:483
typename std::remove_cv< T >::type value_type
Definition: span.h:426
XGBOOST_DEVICE Span< element_type, dynamic_extent > last(std::size_t _count) const
Definition: span.h:583
XGBOOST_DEVICE pointer operator->() const
Definition: span.h:186
XGBOOST_DEVICE SpanIterator & operator--()
Definition: span.h:203
std::random_access_iterator_tag iterator_category
Definition: span.h:157
XGBOOST_DEVICE reference operator*() const
Definition: span.h:178
constexpr XGBOOST_DEVICE friend bool operator<(SpanIterator _lhs, SpanIterator _rhs) __span_noexcept
Definition: span.h:251
typename SpanType::value_type value_type
Definition: span.h:158
constexpr XGBOOST_DEVICE SpanIterator(const SpanType *_span, typename SpanType::index_type _idx) __span_noexcept
Definition: span.h:167
constexpr XGBOOST_DEVICE friend bool operator>(SpanIterator _lhs, SpanIterator _rhs) __span_noexcept
Definition: span.h:261
constexpr XGBOOST_DEVICE SpanIterator(const SpanIterator< SpanType, B > &other_) __span_noexcept
Definition: span.h:174
XGBOOST_DEVICE SpanIterator operator++(int)
Definition: span.h:197
constexpr XGBOOST_DEVICE friend bool operator==(SpanIterator _lhs, SpanIterator _rhs) __span_noexcept
Definition: span.h:241
XGBOOST_DEVICE SpanIterator & operator+=(difference_type n)
Definition: span.h:220
XGBOOST_DEVICE SpanIterator operator-(difference_type n) const
Definition: span.h:231
XGBOOST_DEVICE SpanIterator & operator-=(difference_type n)
Definition: span.h:236
detail::ptrdiff_t difference_type
Definition: span.h:159
typename std::conditional< IsConst, const ElementType, ElementType >::type & reference
Definition: span.h:162
XGBOOST_DEVICE reference operator[](difference_type n) const
Definition: span.h:182
typename std::add_pointer< reference >::type pointer
Definition: span.h:163
XGBOOST_DEVICE difference_type operator-(SpanIterator rhs) const
Definition: span.h:226
const SpanType * span_
Definition: span.h:272
constexpr XGBOOST_DEVICE friend bool operator>=(SpanIterator _lhs, SpanIterator _rhs) __span_noexcept
Definition: span.h:266
XGBOOST_DEVICE SpanIterator operator+(difference_type n) const
Definition: span.h:215
constexpr SpanIterator()=default
constexpr XGBOOST_DEVICE friend bool operator!=(SpanIterator _lhs, SpanIterator _rhs) __span_noexcept
Definition: span.h:246
constexpr XGBOOST_DEVICE friend bool operator<=(SpanIterator _lhs, SpanIterator _rhs) __span_noexcept
Definition: span.h:256
XGBOOST_DEVICE SpanIterator operator--(int)
Definition: span.h:209
SpanType::index_type index_
Definition: span.h:273
XGBOOST_DEVICE SpanIterator & operator++()
Definition: span.h:191
XGBOOST_DEVICE bool LexicographicalCompare(InputIt1 first1, InputIt1 last1, InputIt2 first2, InputIt2 last2)
Definition: span.h:337
typename std::conditional< std::is_same< std::ptrdiff_t, std::int64_t >::value, std::ptrdiff_t, std::int64_t >::type ptrdiff_t
Definition: span.h:135
constexpr XGBOOST_DEVICE bool operator>=(Span< T, X > l, Span< U, Y > r)
Definition: span.h:656
constexpr XGBOOST_DEVICE bool operator!=(Span< T, X > l, Span< U, Y > r)
Definition: span.h:632
constexpr std::size_t dynamic_extent
Definition: span.h:142
byte
Definition: span.h:145
XGBOOST_DEVICE auto as_writable_bytes(Span< T, E > s) __span_noexcept -> Span< byte, detail::ExtentAsBytesValue< T, E >::value >
Definition: span.h:667
XGBOOST_DEVICE bool operator==(Span< T, X > l, Span< U, Y > r)
Definition: span.h:618
XGBOOST_DEVICE auto as_bytes(Span< T, E > s) __span_noexcept -> Span< const byte, detail::ExtentAsBytesValue< T, E >::value >
Definition: span.h:661
constexpr XGBOOST_DEVICE bool operator>(Span< T, X > l, Span< U, Y > r)
Definition: span.h:648
constexpr XGBOOST_DEVICE bool operator<=(Span< T, X > l, Span< U, Y > r)
Definition: span.h:643
constexpr XGBOOST_DEVICE bool operator<(Span< T, X > l, Span< U, Y > r)
Definition: span.h:637
constexpr size_t Offset(S(&strides)[D], size_t n, Head head)
Definition: linalg.h:48
namespace of xgboost
Definition: base.h:110
#define SPAN_LT(lhs, rhs)
Definition: span.h:124
#define __span_noexcept
span class based on ISO++20 span
Definition: span.h:71
#define SPAN_CHECK(cond)
Definition: span.h:118
constexpr XGBOOST_DEVICE bool operator()(const T &_x, const T &_y) const
Definition: span.h:329
constexpr XGBOOST_DEVICE bool operator()(const T &_x, const T &_y) const
Definition: span.h:322