microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-13 01:36:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

make build pass

Browse Source
This commit is contained in:
Binyang Li
2023-04-21 09:41:52 +00:00
parent 7ac861b1e9
commit 7e1a77a132
4 changed files with 598 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/include/bootstrap.h
View File

@@ -18,11 +18,13 @@ class MscclppBootstrap : Bootstrap {
public:
MscclppBootstrap(std::string ipPortPair, int rank, int nRanks);
MscclppBootstrap(mscclppBootstrapHandle handle, int rank, int nRanks);
~MscclppBootstrap() = default;
void Initialize();
void Send(void* data, int size, int peer, int tag);
void Recv(void* data, int size, int peer, int tag);
void AllGather(void* allData, int size);
void Barrier();
void Send(void* data, int size, int peer, int tag) override;
void Recv(void* data, int size, int peer, int tag) override;
void AllGather(void* allData, int size) override;
void Barrier() override;
void Close();
struct UniqueId;
UniqueId GetUniqueId();
@@ -32,6 +34,7 @@ private:
std::unique_ptr<Impl> pimpl_;
};
mscclppResult_t bootstrapNetInit(const char* ip_port_pair = NULL);
mscclppResult_t bootstrapCreateRoot(struct mscclppBootstrapHandle* handle);
mscclppResult_t bootstrapGetUniqueId(struct mscclppBootstrapHandle* handle, bool isRoot = true,
const char* ip_port_pair = NULL);

2
src/include/comm.h
View File

@@ -44,7 +44,7 @@ struct mscclppComm
struct mscclppDevConn devConns[MAXCONNECTIONS];
int nConns;
MscclppBootstrap bootstrap;
void* bootstrap;
// Magic number for all network communication. Not a security key -- only goal is to detect mismatches.
uint64_t magic;