bootstrap.hpp

 
#pragma once
 
#include <chrono>
#include <memory>
#include <optional>
#include <string>
#include <string_view>
 
#include <rapidsmpf/bootstrap/backend.hpp>
#include <rapidsmpf/bootstrap/types.hpp>
 
namespace rapidsmpf::bootstrap {
 
struct Context {
    Rank rank;
 
    Rank nranks;
 
    BackendType type;
 
    std::optional<std::string> coord_dir;
 
    std::shared_ptr<detail::Backend> backend;
};
 
Context init(BackendType type = BackendType::AUTO);
 
void barrier(Context const& ctx);
 
void sync(Context const& ctx);
 
void put(Context const& ctx, std::string const& key, std::string_view value);
 
std::string get(
    Context const& ctx,
    std::string const& key,
    Duration timeout = std::chrono::seconds{30}
);
 
}  // namespace rapidsmpf::bootstrap