xgboost
Public Member Functions | Static Public Member Functions | Friends | List of all members
xgboost::GPUDistribution Class Reference

#include <host_device_vector.h>

Collaboration diagram for xgboost::GPUDistribution:
Collaboration graph

Public Member Functions

 GPUDistribution (GPUSet devices=GPUSet::Empty())
 
GPUSet Devices () const
 
bool IsEmpty () const
 
size_t ShardStart (size_t size, int index) const
 
size_t ShardSize (size_t size, size_t index) const
 
size_t ShardProperSize (size_t size, size_t index) const
 
bool IsFixedSize () const
 

Static Public Member Functions

static GPUDistribution Empty ()
 
static GPUDistribution Block (GPUSet devices)
 
static GPUDistribution Overlap (GPUSet devices, int overlap)
 
static GPUDistribution Granular (GPUSet devices, int granularity)
 
static GPUDistribution Explicit (GPUSet devices, std::vector< size_t > offsets)
 

Friends

template<typename T >
struct HostDeviceVectorImpl
 
bool operator== (const GPUDistribution &a, const GPUDistribution &b)
 
bool operator!= (const GPUDistribution &a, const GPUDistribution &b)
 

Constructor & Destructor Documentation

◆ GPUDistribution()

xgboost::GPUDistribution::GPUDistribution ( GPUSet  devices = GPUSet::Empty())
inlineexplicit

Member Function Documentation

◆ Block()

static GPUDistribution xgboost::GPUDistribution::Block ( GPUSet  devices)
inlinestatic

◆ Devices()

GPUSet xgboost::GPUDistribution::Devices ( ) const
inline

◆ Empty()

static GPUDistribution xgboost::GPUDistribution::Empty ( )
inlinestatic

◆ Explicit()

static GPUDistribution xgboost::GPUDistribution::Explicit ( GPUSet  devices,
std::vector< size_t >  offsets 
)
inlinestatic

◆ Granular()

static GPUDistribution xgboost::GPUDistribution::Granular ( GPUSet  devices,
int  granularity 
)
inlinestatic

◆ IsEmpty()

bool xgboost::GPUDistribution::IsEmpty ( ) const
inline

◆ IsFixedSize()

bool xgboost::GPUDistribution::IsFixedSize ( ) const
inline

◆ Overlap()

static GPUDistribution xgboost::GPUDistribution::Overlap ( GPUSet  devices,
int  overlap 
)
inlinestatic

◆ ShardProperSize()

size_t xgboost::GPUDistribution::ShardProperSize ( size_t  size,
size_t  index 
) const
inline

◆ ShardSize()

size_t xgboost::GPUDistribution::ShardSize ( size_t  size,
size_t  index 
) const
inline

◆ ShardStart()

size_t xgboost::GPUDistribution::ShardStart ( size_t  size,
int  index 
) const
inline

Friends And Related Function Documentation

◆ HostDeviceVectorImpl

template<typename T >
friend struct HostDeviceVectorImpl
friend

◆ operator!=

bool operator!= ( const GPUDistribution a,
const GPUDistribution b 
)
friend

◆ operator==

bool operator== ( const GPUDistribution a,
const GPUDistribution b 
)
friend

The documentation for this class was generated from the following file: