Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Data Fields
LightGBM::BruckMap Class Reference

The network structure for all_gather. More...

#include <network.h>

Public Member Functions

 BruckMap (int n)
 

Static Public Member Functions

static BruckMap Construct (int rank, int num_machines)
 Create the object of bruck map.
 

Data Fields

int k
 The communication times for one all gather operation.
 
std::vector< int > in_ranks
 in_ranks[i] means the incomming rank on i-th communication
 
std::vector< int > out_ranks
 out_ranks[i] means the out rank on i-th communication
 

Detailed Description

The network structure for all_gather.

Member Function Documentation

◆ Construct()

BruckMap LightGBM::BruckMap::Construct ( int  rank,
int  num_machines 
)
static

Create the object of bruck map.

Parameters
rankRank of this machine
num_machinesThe total number of machines
Returns
The object of bruck map

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