49 #ifndef XGBOOST_HOST_DEVICE_VECTOR_H_
50 #define XGBOOST_HOST_DEVICE_VECTOR_H_
55 #include <initializer_list>
56 #include <type_traits>
64 void SetCudaSetDeviceHandler(
void (*handler)(
int));
88 static_assert(std::is_standard_layout<T>::value,
"HostDeviceVector admits only POD types");
102 [[nodiscard]]
bool Empty()
const {
return Size() == 0; }
103 [[nodiscard]] std::size_t
Size()
const;
121 void Copy(
const std::vector<T>& other);
122 void Copy(std::initializer_list<T> other);
Definition: host_device_vector.h:87
const std::vector< T > & HostVector() const
Definition: host_device_vector.h:128
HostDeviceVector(size_t size=0, T v=T(), DeviceOrd device=DeviceOrd::CPU())
const T * ConstDevicePointer() const
void Extend(const HostDeviceVector< T > &other)
HostDeviceVector(const HostDeviceVector< T > &)=delete
void Copy(const HostDeviceVector< T > &other)
bool Empty() const
Definition: host_device_vector.h:102
common::Span< T const > ConstHostSpan() const
Definition: host_device_vector.h:115
std::vector< T > & HostVector()
common::Span< T const > HostSpan() const
Definition: host_device_vector.h:114
bool DeviceCanWrite() const
void Copy(std::initializer_list< T > other)
HostDeviceVector< T > & operator=(const HostDeviceVector< T > &)=delete
HostDeviceVector(std::initializer_list< T > init, DeviceOrd device=DeviceOrd::CPU())
void Resize(size_t new_size, T v=T())
T value_type
Definition: host_device_vector.h:140
HostDeviceVector(HostDeviceVector< T > &&)
common::Span< const T > ConstDeviceSpan() const
const std::vector< T > & ConstHostVector() const
T * HostPointer()
Definition: host_device_vector.h:112
HostDeviceVector(const std::vector< T > &init, DeviceOrd device=DeviceOrd::CPU())
const T * HostPointer() const
Definition: host_device_vector.h:117
common::Span< T > DeviceSpan()
bool DeviceCanRead() const
common::Span< T > HostSpan()
Definition: host_device_vector.h:113
common::Span< const T > DeviceSpan() const
Definition: host_device_vector.h:107
void Copy(const std::vector< T > &other)
GPUAccess DeviceAccess() const
HostDeviceVector< T > & operator=(HostDeviceVector< T > &&)
const T * DevicePointer() const
Definition: host_device_vector.h:110
void SetDevice(DeviceOrd device) const
bool HostCanWrite() const
const T * ConstHostPointer() const
Definition: host_device_vector.h:116
span class implementation, based on ISO++20 span<T>. The interface should be the same.
Definition: span.h:422
Core data structure for multi-target trees.
Definition: base.h:87
GPUAccess
Controls data access from the GPU.
Definition: host_device_vector.h:80
@ kNone
Definition: host_device_vector.h:81
@ kRead
Definition: host_device_vector.h:81
@ kWrite
Definition: host_device_vector.h:83
A type for device ordinal. The type is packed into 32-bit for efficient use in viewing types like lin...
Definition: context.h:34
constexpr static auto CPU()
Constructor for CPU.
Definition: context.h:64
Definition: host_device_vector.h:67