xgboost::common::ParallelGroupBuilder< ValueType, SizeType > Class Template Reference

multi-thread version of group builder More...

#include <group_data.h>

Collaboration diagram for xgboost::common::ParallelGroupBuilder< ValueType, SizeType >:

Public Member Functions
	ParallelGroupBuilder (std::vector< SizeType > *p_rptr, std::vector< ValueType > *p_data, size_t base_row_offset=0)
	parallel group builder of data. More...
	ParallelGroupBuilder (std::vector< SizeType > *p_rptr, std::vector< ValueType > *p_data, std::vector< std::vector< SizeType > > *p_thread_rptr, size_t base_row_offset=0)
void	InitBudget (std::size_t max_key, int nthread)
	step 1: initialize the helper, with hint of number keys and thread used in the construction More...
void	AddBudget (std::size_t key, int threadid, SizeType nelem=1)
	step 2: add budget to each key More...
void	InitStorage ()
	step 3: initialize the necessary storage More...
void	Push (std::size_t key, ValueType value, int threadid)
	step 4: add data to the allocated space, the calls to this function should be exactly match previous call to AddBudget More...

Detailed Description

template<typename ValueType, typename SizeType = bst_ulong>
class xgboost::common::ParallelGroupBuilder< ValueType, SizeType >

multi-thread version of group builder

Template Parameters

ValueType	type of entries in the sparse matrix
SizeType	type of the index range holder

Constructor & Destructor Documentation

ParallelGroupBuilder() [1/2]

template<typename ValueType , typename SizeType = bst_ulong>

xgboost::common::ParallelGroupBuilder< ValueType, SizeType >::ParallelGroupBuilder ( std::vector< SizeType > *  p_rptr, std::vector< ValueType > *  p_data, size_t  base_row_offset = 0  )

inline

parallel group builder of data.

Parameters

[in,out]	p_rptr	Row offsets for CSR matrix.
[in,out]	p_data	Data vector to populate
	base_row_offset	(Optional) If the matrix we are building is already partially populated, use this to indicate the row index we are starting from. This saves considerable amounts of time/memory when incrementaly building.

ParallelGroupBuilder() [2/2]

template<typename ValueType , typename SizeType = bst_ulong>

xgboost::common::ParallelGroupBuilder< ValueType, SizeType >::ParallelGroupBuilder ( std::vector< SizeType > *  p_rptr, std::vector< ValueType > *  p_data, std::vector< std::vector< SizeType > > *  p_thread_rptr, size_t  base_row_offset = 0  )

inline

Member Function Documentation

AddBudget()

template<typename ValueType , typename SizeType = bst_ulong>

void xgboost::common::ParallelGroupBuilder< ValueType, SizeType >::AddBudget ( std::size_t  key, int  threadid, SizeType  nelem = 1  )

inline

step 2: add budget to each key

Parameters

key	the key
threadid	the id of thread that calls this function
nelem	number of element budget add to this row

InitBudget()

template<typename ValueType , typename SizeType = bst_ulong>

void xgboost::common::ParallelGroupBuilder< ValueType, SizeType >::InitBudget ( std::size_t  max_key, int  nthread  )

inline

step 1: initialize the helper, with hint of number keys and thread used in the construction

Parameters

max_key	number of keys in the matrix, can be smaller than expected
nthread	number of thread that will be used in construction

InitStorage()

template<typename ValueType , typename SizeType = bst_ulong>

void xgboost::common::ParallelGroupBuilder< ValueType, SizeType >::InitStorage ( )

inline

step 3: initialize the necessary storage

Push()

template<typename ValueType , typename SizeType = bst_ulong>

void xgboost::common::ParallelGroupBuilder< ValueType, SizeType >::Push ( std::size_t  key, ValueType  value, int  threadid  )

inline

step 4: add data to the allocated space, the calls to this function should be exactly match previous call to AddBudget

Parameters

key	the key of group.
value	The value to be pushed to the group.
threadid	the id of thread that calls this function

---

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

• /workspace/src/common/group_data.h