mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-13 01:36:10 +00:00
ae56698d67
More intuitive interfaces for creating semaphores and channels. Also allows channel construction using third-party bootstrappers directly without overriding MSCCL++ Bootstrap.
116 lines
3.9 KiB
C++
116 lines
3.9 KiB
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT license.
|
|
|
|
#include <mscclpp/core.hpp>
|
|
#include <sstream>
|
|
|
|
#include "api.h"
|
|
|
|
namespace mscclpp {
|
|
|
|
MSCCLPP_API_CPP std::string version() {
|
|
std::stringstream ss;
|
|
ss << MSCCLPP_MAJOR << "." << MSCCLPP_MINOR << "." << MSCCLPP_PATCH;
|
|
return ss.str();
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags::TransportFlags(Transport transport)
|
|
: detail::TransportFlagsBase(1 << static_cast<size_t>(transport)) {}
|
|
|
|
MSCCLPP_API_CPP bool TransportFlags::has(Transport transport) const {
|
|
return detail::TransportFlagsBase::test(static_cast<size_t>(transport));
|
|
}
|
|
|
|
MSCCLPP_API_CPP bool TransportFlags::none() const { return detail::TransportFlagsBase::none(); }
|
|
|
|
MSCCLPP_API_CPP bool TransportFlags::any() const { return detail::TransportFlagsBase::any(); }
|
|
|
|
MSCCLPP_API_CPP bool TransportFlags::all() const { return detail::TransportFlagsBase::all(); }
|
|
|
|
MSCCLPP_API_CPP size_t TransportFlags::count() const { return detail::TransportFlagsBase::count(); }
|
|
|
|
MSCCLPP_API_CPP TransportFlags& TransportFlags::operator|=(TransportFlags other) {
|
|
detail::TransportFlagsBase::operator|=(other);
|
|
return *this;
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags TransportFlags::operator|(TransportFlags other) const {
|
|
return TransportFlags(*this) |= other;
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags TransportFlags::operator|(Transport transport) const {
|
|
return *this | TransportFlags(transport);
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags& TransportFlags::operator&=(TransportFlags other) {
|
|
detail::TransportFlagsBase::operator&=(other);
|
|
return *this;
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags TransportFlags::operator&(TransportFlags other) const {
|
|
return TransportFlags(*this) &= other;
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags TransportFlags::operator&(Transport transport) const {
|
|
return *this & TransportFlags(transport);
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags& TransportFlags::operator^=(TransportFlags other) {
|
|
detail::TransportFlagsBase::operator^=(other);
|
|
return *this;
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags TransportFlags::operator^(TransportFlags other) const {
|
|
return TransportFlags(*this) ^= other;
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags TransportFlags::operator^(Transport transport) const {
|
|
return *this ^ TransportFlags(transport);
|
|
}
|
|
|
|
MSCCLPP_API_CPP TransportFlags TransportFlags::operator~() const { return TransportFlags(*this).flip(); }
|
|
|
|
MSCCLPP_API_CPP bool TransportFlags::operator==(TransportFlags other) const {
|
|
return detail::TransportFlagsBase::operator==(other);
|
|
}
|
|
|
|
MSCCLPP_API_CPP bool TransportFlags::operator!=(TransportFlags other) const {
|
|
return detail::TransportFlagsBase::operator!=(other);
|
|
}
|
|
|
|
MSCCLPP_API_CPP detail::TransportFlagsBase TransportFlags::toBitset() const { return *this; }
|
|
|
|
TransportFlags::TransportFlags(detail::TransportFlagsBase bitset) : detail::TransportFlagsBase(bitset) {}
|
|
|
|
const TransportFlags NoTransports = TransportFlags();
|
|
|
|
const TransportFlags AllIBTransports = Transport::IB0 | Transport::IB1 | Transport::IB2 | Transport::IB3 |
|
|
Transport::IB4 | Transport::IB5 | Transport::IB6 | Transport::IB7;
|
|
|
|
const TransportFlags AllTransports = AllIBTransports | Transport::CudaIpc | Transport::Ethernet;
|
|
|
|
} // namespace mscclpp
|
|
|
|
namespace std {
|
|
|
|
std::string to_string(const mscclpp::Transport& transport) {
|
|
static const std::string TransportNames[] = {"UNK", "IPC", "NVLS", "IB0", "IB1", "IB2", "IB3",
|
|
"IB4", "IB5", "IB6", "IB7", "ETH", "NUM"};
|
|
return TransportNames[static_cast<size_t>(transport)];
|
|
}
|
|
|
|
std::string to_string(const mscclpp::Device& device) {
|
|
std::stringstream ss;
|
|
ss << "Device(type=" << to_string(device.type) << ", id=" << device.id << ")";
|
|
return ss.str();
|
|
}
|
|
|
|
template <>
|
|
struct hash<mscclpp::TransportFlags> {
|
|
size_t operator()(const mscclpp::TransportFlags& flags) const {
|
|
return hash<mscclpp::detail::TransportFlagsBase>()(flags.toBitset());
|
|
}
|
|
};
|
|
|
|
} // namespace std
|