Node Class

[ryanmrichard]

The Node class maps to the concept of an MPI rank.

class Node {
public:

/// True if the current execution context is on this node
bool is_me();

/// Get the number of resource units of type T, e.g. `auto ngpus = node.count<GPU>();`
template<typename T>
size_type count();

/// Gets the i-th resource of type T, e.g. `auto cpu_0 = node.get<CPU>(0);`
template<typename T>
T get(size_type i);

// Point to point
template<typename Container>
void send(Container&& data, Node destination);

template<typename ResultType>
ResultType receive(Node Sender);

// Point to all collectives
template<template <typename> typename Container>
Container broadcast(Container&& data, Cluster destination);

template<template <typename> typename Container>
auto scatter(Container&& data, Cluster destination);

// Standard output stream on this node
std::ostream stdcout ;

// Standard error stream on this node
std::ostream stderr;
};

Notes:

• Can add overloads with a string tag to distinguish messages.
• Streams will need to account for the current rank so only the appropriate node prints