xgboost
|
#include <host_device_vector.h>
Public Member Functions | |
HostDeviceVector (size_t size=0, T v=T(), const GPUDistribution &distribution=GPUDistribution()) | |
HostDeviceVector (std::initializer_list< T > init, const GPUDistribution &distribution=GPUDistribution()) | |
HostDeviceVector (const std::vector< T > &init, const GPUDistribution &distribution=GPUDistribution()) | |
~HostDeviceVector () | |
HostDeviceVector (const HostDeviceVector< T > &) | |
HostDeviceVector< T > & | operator= (const HostDeviceVector< T > &) |
size_t | Size () const |
GPUSet | Devices () const |
const GPUDistribution & | Distribution () const |
common::Span< T > | DeviceSpan (int device) |
common::Span< const T > | ConstDeviceSpan (int device) const |
common::Span< const T > | DeviceSpan (int device) const |
T * | DevicePointer (int device) |
const T * | ConstDevicePointer (int device) const |
const T * | DevicePointer (int device) const |
T * | HostPointer () |
const T * | ConstHostPointer () const |
const T * | HostPointer () const |
size_t | DeviceStart (int device) const |
size_t | DeviceSize (int device) const |
void | Fill (T v) |
void | Copy (const HostDeviceVector< T > &other) |
void | Copy (const std::vector< T > &other) |
void | Copy (std::initializer_list< T > other) |
std::vector< T > & | HostVector () |
const std::vector< T > & | ConstHostVector () const |
const std::vector< T > & | HostVector () const |
bool | HostCanAccess (GPUAccess access) const |
bool | DeviceCanAccess (int device, GPUAccess access) const |
void | Shard (const GPUDistribution &distribution) const |
Specify memory distribution. More... | |
void | Shard (GPUSet devices) const |
void | Reshard (const GPUDistribution &distribution) |
Change memory distribution. More... | |
void | Resize (size_t new_size, T v=T()) |
|
explicit |
xgboost::HostDeviceVector< T >::HostDeviceVector | ( | std::initializer_list< T > | init, |
const GPUDistribution & | distribution = GPUDistribution() |
||
) |
|
explicit |
xgboost::HostDeviceVector< T >::~HostDeviceVector | ( | ) |
xgboost::HostDeviceVector< T >::HostDeviceVector | ( | const HostDeviceVector< T > & | ) |
const T* xgboost::HostDeviceVector< T >::ConstDevicePointer | ( | int | device | ) | const |
common::Span<const T> xgboost::HostDeviceVector< T >::ConstDeviceSpan | ( | int | device | ) | const |
|
inline |
const std::vector<T>& xgboost::HostDeviceVector< T >::ConstHostVector | ( | ) | const |
void xgboost::HostDeviceVector< T >::Copy | ( | const HostDeviceVector< T > & | other | ) |
void xgboost::HostDeviceVector< T >::Copy | ( | const std::vector< T > & | other | ) |
void xgboost::HostDeviceVector< T >::Copy | ( | std::initializer_list< T > | other | ) |
bool xgboost::HostDeviceVector< T >::DeviceCanAccess | ( | int | device, |
GPUAccess | access | ||
) | const |
T* xgboost::HostDeviceVector< T >::DevicePointer | ( | int | device | ) |
|
inline |
GPUSet xgboost::HostDeviceVector< T >::Devices | ( | ) | const |
size_t xgboost::HostDeviceVector< T >::DeviceSize | ( | int | device | ) | const |
common::Span<T> xgboost::HostDeviceVector< T >::DeviceSpan | ( | int | device | ) |
|
inline |
size_t xgboost::HostDeviceVector< T >::DeviceStart | ( | int | device | ) | const |
const GPUDistribution& xgboost::HostDeviceVector< T >::Distribution | ( | ) | const |
void xgboost::HostDeviceVector< T >::Fill | ( | T | v | ) |
bool xgboost::HostDeviceVector< T >::HostCanAccess | ( | GPUAccess | access | ) | const |
|
inline |
|
inline |
std::vector<T>& xgboost::HostDeviceVector< T >::HostVector | ( | ) |
|
inline |
HostDeviceVector<T>& xgboost::HostDeviceVector< T >::operator= | ( | const HostDeviceVector< T > & | ) |
void xgboost::HostDeviceVector< T >::Reshard | ( | const GPUDistribution & | distribution | ) |
Change memory distribution.
void xgboost::HostDeviceVector< T >::Resize | ( | size_t | new_size, |
T | v = T() |
||
) |
void xgboost::HostDeviceVector< T >::Shard | ( | const GPUDistribution & | distribution | ) | const |
Specify memory distribution.
void xgboost::HostDeviceVector< T >::Shard | ( | GPUSet | devices | ) | const |
size_t xgboost::HostDeviceVector< T >::Size | ( | ) | const |