bootstrap.hpp

 
#pragma once
 
#include <chrono>
#include <cstdint>
#include <memory>
#include <optional>
#include <string>
 
#include <rapidsmpf/config.hpp>
#include <rapidsmpf/utils.hpp>
 
namespace rapidsmpf::bootstrap {
 
using Rank = std::int32_t;
 
using rapidsmpf::Duration;
 
enum class Backend {
    AUTO,
 
    FILE,
};
 
struct Context {
    Rank rank;
 
    Rank nranks;
 
    Backend backend;
 
    std::optional<std::string> coord_dir;
};
 
Context init(Backend backend = Backend::AUTO);
 
void broadcast(Context const& ctx, void* data, std::size_t size, Rank root = 0);
 
void barrier(Context const& ctx);
 
void put(Context const& ctx, std::string const& key, std::string const& value);
 
std::string get(
    Context const& ctx,
    std::string const& key,
    Duration timeout = std::chrono::seconds{30}
);
 
}  // namespace rapidsmpf::bootstrap