Commit d49b9079 authored by Marc Vef's avatar Marc Vef
Browse files

Adding UCX and OFI protocols

parent 9a8dc12e

include/common/common_defs.hpp

+8 −2
Original line number Diff line number Diff line
@@ -58,11 +58,17 @@ constexpr auto get_chunk_stat = "rpc_srv_chunk_stat"; 
} // namespace tag



namespace protocol {

constexpr auto ofi_psm2 = "ofi+psm2";

constexpr auto na_sm = "na+sm";

constexpr auto ofi_sockets = "ofi+sockets";

constexpr auto ofi_tcp = "ofi+tcp";

constexpr auto ofi_verbs = "ofi+verbs";

constexpr auto na_sm = "na+sm";

constexpr auto ofi_psm2 = "ofi+psm2";

constexpr auto ucx_all = "ucx+all";

constexpr auto ucx_tcp = "ucx+tcp";

constexpr auto ucx_rc = "ucx+rc";

constexpr auto ucx_ud = "ucx+ud";

const auto all_protocols = {na_sm,   ofi_sockets, ofi_tcp, ofi_verbs, ofi_psm2,

                            ucx_all, ucx_tcp,     ucx_rc,  ucx_ud};

} // namespace protocol

} // namespace gkfs::rpc

src/daemon/daemon.cpp

+8 −4
Original line number Diff line number Diff line
@@ -496,14 +496,18 @@ parse_input(const cli_options& opts, const CLI::App& desc) { 
    auto rpc_protocol = string(gkfs::rpc::protocol::ofi_sockets);

    if(desc.count("--rpc-protocol")) {

        rpc_protocol = opts.rpc_protocol;

        if(rpc_protocol != gkfs::rpc::protocol::ofi_verbs &&

           rpc_protocol != gkfs::rpc::protocol::ofi_sockets &&

           rpc_protocol != gkfs::rpc::protocol::ofi_psm2) {

        auto protocol_found = false;

        for(const auto& valid_protocol : gkfs::rpc::protocol::all_protocols) {

            if(rpc_protocol == valid_protocol) {

                protocol_found = true;

                break;

            }

        }

        if(!protocol_found)

            throw runtime_error(fmt::format(

                    "Given RPC protocol '{}' not supported. Check --help for supported protocols.",

                    rpc_protocol));

    }

    }



    auto use_auto_sm = desc.count("--auto-sm") != 0;

    GKFS_DATA->use_auto_sm(use_auto_sm);