init.cpp File Reference

#include "config/config.hpp"
#include "init.hpp"
#include "utils.hpp"
#include "communication.hpp"
#include <utils/mpi/gather_buffer.hpp>
#include <mpi.h>
#include <algorithm>
#include <iterator>
#include <set>
#include <string>
#include <utility>
#include <vector>

Include dependency graph for cuda/init.cpp:

Go to the source code of this file.

Classes
struct	CompareDevices
	Helper class for device sets. More...

Functions
std::vector< EspressoGpuDevice >	cuda_gather_gpus ()
	Gather list of CUDA devices from all nodes on the head node.
void	cuda_on_program_start ()
	Called on program start.

Function Documentation

cuda_gather_gpus()

std::vector< EspressoGpuDevice > cuda_gather_gpus

(

)

Gather list of CUDA devices from all nodes on the head node.

Gather unique list of CUDA devices on all nodes.

It relies on MPI_Get_processor_name() to get a unique identifier of the physical node, as opposed to the logical rank of which there can be more than one per node.

Definition at line 56 of file cuda/init.cpp.

References comm_cart, cuda_get_device_props(), cuda_get_n_gpus(), Utils::Mpi::gather_buffer(), invoke_skip_cuda_exceptions(), stream, and this_node.

Referenced by ScriptInterface::System::CudaInitHandle::do_call_method().

cuda_on_program_start()

void cuda_on_program_start

(

)

Called on program start.

Definition at line 91 of file cuda/init.cpp.

References communicator, cuda_init(), invoke_skip_cuda_exceptions(), and this_node.

Referenced by CommunicationEnvironment::CommunicationEnvironment().