device.h File Reference

#include "types.h"
#include <fstream>
#include "mpi.h"

Include dependency graph for device.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	base_device
namespace	base_device::information

Functions
template<typename Device >
base_device::AbacusDevice_t	base_device::get_device_type (const Device *dev)
template<typename T >
std::string	base_device::get_current_precision (const T *var)
std::string	base_device::information::get_device_name (std::string device_flag)
	Get the device name for source_esolver.
int	base_device::information::get_device_num (std::string device_flag)
	Get the device number for source_esolver.
void	base_device::information::output_device_info (std::ostream &output)
	Output the device information 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.
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_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)
int	base_device::information::stringCmp (const void *a, const void *b)
template<typename Device >
void	base_device::information::print_device_info (const Device *dev, std::ofstream &ofs_device)
template<typename Device >
void	base_device::information::record_device_memory (const Device *dev, std::ofstream &ofs_device, std::string str, size_t size)