ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
Loading...
Searching...
No Matches
Functions
base_device::information Namespace Reference

Functions

int stringCmp (const void *a, const void *b)
 
int get_node_rank ()
 Get the rank of current node Note that GPU can only be binded with CPU in the same node.
 
int get_node_rank_with_mpi_shared (const MPI_Comm mpi_comm)
 
std::string get_device_flag (const std::string &device, const std::string &basis_type)
 Get the device flag object for source_io PARAM.inp.device.
 
int get_device_kpar (const int &kpar, const int &bndpar)
 Get the device kpar object for source_io GlobalV::KPAR.
 
std::string get_device_info (std::string device_flag)
 Get the device info object for source_esolver.
 
template<typename Device >
void print_device_info (const Device *dev, std::ofstream &ofs_device)
 
template<typename Device >
void record_device_memory (const Device *dev, std::ofstream &ofs_device, std::string str, size_t size)
 

Function Documentation

◆ get_device_flag()

std::string base_device::information::get_device_flag ( const std::string &  device,
const std::string &  basis_type 
)

Get the device flag object for source_io PARAM.inp.device.

Here is the call graph for this function:
Here is the caller graph for this function:

◆ get_device_info()

std::string base_device::information::get_device_info ( std::string  device_flag)

Get the device info object for source_esolver.

Here is the caller graph for this function:

◆ get_device_kpar()

int base_device::information::get_device_kpar ( const int &  kpar,
const int &  bndpar 
)

Get the device kpar object for source_io GlobalV::KPAR.

Here is the call graph for this function:
Here is the caller graph for this function:

◆ get_node_rank()

int base_device::information::get_node_rank ( )

Get the rank of current node Note that GPU can only be binded with CPU in the same node.

Returns
int
Here is the call graph for this function:
Here is the caller graph for this function:

◆ get_node_rank_with_mpi_shared()

int base_device::information::get_node_rank_with_mpi_shared ( const MPI_Comm  mpi_comm)

◆ print_device_info()

template<typename Device >
void base_device::information::print_device_info ( const Device *  dev,
std::ofstream &  ofs_device 
)

◆ record_device_memory()

template<typename Device >
void base_device::information::record_device_memory ( const Device *  dev,
std::ofstream &  ofs_device,
std::string  str,
size_t  size 
)

◆ stringCmp()

int base_device::information::stringCmp ( const void *  a,
const void *  b 
)
Here is the caller graph for this function: