HMLP: High-performance Machine Learning Primitives
Public Types | Public Member Functions | Public Attributes | List of all members
hmlp::mpitree::Setup< SPLITTER, DATATYPE > Class Template Reference

Data and setup that are shared with all nodes. More...

#include <tree_mpi.hpp>

Public Types

typedef DATATYPE T
 

Public Member Functions

vector< size_t > ContainAny (vector< size_t > &queries, size_t target)
 Check if this node contain any query using morton. Notice that queries[] contains gids; thus, morton[] needs to be accessed using gids. More...
 

Public Attributes

size_t m
 
size_t max_depth = 15
 
DistData< STAR, CBLK, pair< T, size_t > > * NN_cblk = NULL
 
DistData< STAR, CIDS, pair< T, size_t > > * NN = NULL
 
vector< size_t > morton
 
SPLITTER splitter
 

Detailed Description

template<typename SPLITTER, typename DATATYPE>
class hmlp::mpitree::Setup< SPLITTER, DATATYPE >

Data and setup that are shared with all nodes.

end class DistSplitTask

Member Function Documentation

template<typename SPLITTER, typename DATATYPE>
vector<size_t> hmlp::mpitree::Setup< SPLITTER, DATATYPE >::ContainAny ( vector< size_t > &  queries,
size_t  target 
)
inline

Check if this node contain any query using morton. Notice that queries[] contains gids; thus, morton[] needs to be accessed using gids.

notice that setup->morton only contains local morton ids

Member Data Documentation

template<typename SPLITTER, typename DATATYPE>
size_t hmlp::mpitree::Setup< SPLITTER, DATATYPE >::m

end ContainAny() maximum leaf node size

template<typename SPLITTER, typename DATATYPE>
size_t hmlp::mpitree::Setup< SPLITTER, DATATYPE >::max_depth = 15

by default we use 4 bits = 0-15 levels

template<typename SPLITTER, typename DATATYPE>
vector<size_t> hmlp::mpitree::Setup< SPLITTER, DATATYPE >::morton

morton ids

template<typename SPLITTER, typename DATATYPE>
DistData<STAR, CBLK, pair<T, size_t> >* hmlp::mpitree::Setup< SPLITTER, DATATYPE >::NN_cblk = NULL

coordinates (accessed with gids) neighbors<distance, gid> (accessed with gids)

template<typename SPLITTER, typename DATATYPE>
SPLITTER hmlp::mpitree::Setup< SPLITTER, DATATYPE >::splitter

tree splitter

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