device.cpp File Reference

#include "device.h"
#include "source_base/tool_quit.h"
#include <base/macros/macros.h>
#include <cstring>
#include <iostream>
#include "mpi.h"

Include dependency graph for device.cpp:

Namespaces
namespace	base_device
namespace	base_device::information

Functions
template<>
base_device::AbacusDevice_t	base_device::get_device_type< base_device::DEVICE_CPU > (const base_device::DEVICE_CPU *dev)
template<>
base_device::AbacusDevice_t	base_device::get_device_type< base_device::DEVICE_GPU > (const base_device::DEVICE_GPU *dev)
template<>
std::string	base_device::get_current_precision (const float *var)
template<>
std::string	base_device::get_current_precision (const double *var)
template<>
std::string	base_device::get_current_precision (const std::complex< float > *var)
template<>
std::string	base_device::get_current_precision (const std::complex< double > *var)
int	base_device::information::stringCmp (const void *a, const void *b)
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)
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.
int	base_device::information::get_device_kpar (const int &kpar, const int &bndpar)
	Get the device kpar object for source_io GlobalV::KPAR.