49 #ifndef XGBOOST_HOST_DEVICE_VECTOR_H_ 50 #define XGBOOST_HOST_DEVICE_VECTOR_H_ 52 #include <initializer_list> 62 void SetCudaSetDeviceHandler(
void (*handler)(
int));
116 void Copy(
const std::vector<T>& other);
117 void Copy(std::initializer_list<T> other);
131 void Resize(
size_t new_size, T v = T());
141 #endif // XGBOOST_HOST_DEVICE_VECTOR_H_ void Copy(const HostDeviceVector< T > &other)
bst_float value_type
Definition: host_device_vector.h:133
HostDeviceVector< T > & operator=(const HostDeviceVector< T > &)=delete
const T * DevicePointer() const
Definition: host_device_vector.h:105
Definition: host_device_vector.h:85
GPUAccess DeviceAccess() const
T * HostPointer()
Definition: host_device_vector.h:107
const T * HostPointer() const
Definition: host_device_vector.h:112
common::Span< const T > ConstDeviceSpan() const
const T * ConstHostPointer() const
Definition: host_device_vector.h:111
const std::vector< T > & HostVector() const
Definition: host_device_vector.h:121
span class implementation, based on ISO++20 span<T>. The interface should be the same.
Definition: span.h:126
Definition: host_device_vector.h:79
Definition: host_device_vector.h:65
Definition: host_device_vector.h:81
common::Span< T const > HostSpan() const
Definition: host_device_vector.h:109
const T * ConstDevicePointer() const
common::Span< const T > DeviceSpan() const
Definition: host_device_vector.h:102
Definition: host_device_vector.h:79
common::Span< T > DeviceSpan()
HostDeviceVector(size_t size=0, T v=T(), int device=-1)
void SetDevice(int device) const
std::vector< T > & HostVector()
namespace of xgboost
Definition: base.h:102
const std::vector< T > & ConstHostVector() const
GPUAccess
Controls data access from the GPU.
Definition: host_device_vector.h:78
bool HostCanWrite() const
void Resize(size_t new_size, T v=T())
bool DeviceCanRead() const
bool DeviceCanWrite() const
common::Span< T > HostSpan()
Definition: host_device_vector.h:108
common::Span< T const > ConstHostSpan() const
Definition: host_device_vector.h:110