microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-27 08:24:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

more tests for bootstrap

Browse Source
This commit is contained in:
Saeed Maleki
2023-04-25 22:26:48 +00:00
parent b73b0132ba
commit 8fc822c848
3 changed files with 133 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/include/mscclpp.hpp
View File

@@ -30,6 +30,8 @@ class BaseBootstrap
public:
BaseBootstrap(){};
virtual ~BaseBootstrap() = default;
virtual int getRank() = 0;
virtual int getNranks() = 0;
virtual void send(void* data, int size, int peer, int tag) = 0;
virtual void recv(void* data, int size, int peer, int tag) = 0;
virtual void allGather(void* allData, int size) = 0;
@@ -47,6 +49,8 @@ public:
void initialize(UniqueId uniqueId);
void initialize(std::string ipPortPair);
int getRank() override;
int getNranks() override;
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;