ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
|
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) |
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.
std::string base_device::information::get_device_info | ( | std::string | device_flag | ) |
Get the device info object for source_esolver.
int base_device::information::get_device_kpar | ( | const int & | kpar, |
const int & | bndpar | ||
) |
Get the device kpar object for source_io GlobalV::KPAR.
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.
int base_device::information::get_node_rank_with_mpi_shared | ( | const MPI_Comm | mpi_comm | ) |
void base_device::information::print_device_info | ( | const Device * | dev, |
std::ofstream & | ofs_device | ||
) |
void base_device::information::record_device_memory | ( | const Device * | dev, |
std::ofstream & | ofs_device, | ||
std::string | str, | ||
size_t | size | ||
) |
int base_device::information::stringCmp | ( | const void * | a, |
const void * | b | ||
) |