Go to the documentation of this file.
49 #ifndef XGBOOST_HOST_DEVICE_VECTOR_H_
50 #define XGBOOST_HOST_DEVICE_VECTOR_H_
52 #include <initializer_list>
54 #include <type_traits>
63 void SetCudaSetDeviceHandler(
void (*handler)(
int));
87 static_assert(std::is_standard_layout<T>::value,
"HostDeviceVector admits only POD types");
120 void Copy(
const std::vector<T>& other);
121 void Copy(std::initializer_list<T> other);
137 void Resize(
size_t new_size, T v = T());
147 #endif // XGBOOST_HOST_DEVICE_VECTOR_H_
const std::vector< T > & HostVector() const
Definition: host_device_vector.h:127
@ kWrite
Definition: host_device_vector.h:82
void Extend(const HostDeviceVector< T > &other)
common::Span< T const > HostSpan() const
Definition: host_device_vector.h:113
common::Span< const T > DeviceSpan() const
Definition: host_device_vector.h:106
Definition: host_device_vector.h:86
common::Span< T > DeviceSpan()
GPUAccess DeviceAccess() const
bool Empty() const
Definition: host_device_vector.h:101
common::Span< T const > ConstHostSpan() const
Definition: host_device_vector.h:114
bool DeviceCanRead() const
bool DeviceCanWrite() const
void SetDevice(int device) const
std::vector< T > & HostVector()
const T * DevicePointer() const
Definition: host_device_vector.h:109
common::Span< T > HostSpan()
Definition: host_device_vector.h:112
GPUAccess
Controls data access from the GPU.
Definition: host_device_vector.h:79
@ kRead
Definition: host_device_vector.h:80
void Copy(const HostDeviceVector< T > &other)
T * HostPointer()
Definition: host_device_vector.h:111
void Resize(size_t new_size, T v=T())
const T * ConstDevicePointer() const
@ kNone
Definition: host_device_vector.h:80
bst_float value_type
Definition: host_device_vector.h:139
const T * ConstHostPointer() const
Definition: host_device_vector.h:115
HostDeviceVector(size_t size=0, T v=T(), int device=-1)
const std::vector< T > & ConstHostVector() const
Definition: host_device_vector.h:66
const T * HostPointer() const
Definition: host_device_vector.h:116
common::Span< const T > ConstDeviceSpan() const
bool HostCanWrite() const
namespace of xgboost
Definition: base.h:110
HostDeviceVector< T > & operator=(const HostDeviceVector< T > &)=delete