|
ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
|
#include "device.h"#include "source_base/tool_quit.h"#include <base/macros/macros.h>#include <cstring>#include <iostream>#include <string>#include "mpi.h"Namespaces | |
| namespace | base_device |
| namespace | base_device::information |
Functions | |
| int | base_device::information::get_node_rank_with_mpi_shared (const MPI_Comm mpi_comm=MPI_COMM_WORLD) |
| Get the local rank within the node using MPI_COMM_TYPE_SHARED. | |
| bool | base_device::information::probe_gpu_availability () |
| Safely probes for GPU availability without exiting on error. | |
| 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. | |