ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
Loading...
Searching...
No Matches
Namespaces | Functions
device.cpp File Reference
#include "device.h"
#include "source_base/tool_quit.h"
#include <base/macros/macros.h>
#include <cstring>
#include <iostream>
#include <string>
#include "mpi.h"
Include dependency graph for device.cpp:

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.