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

wip

Browse Source
This commit is contained in:
Saeed Maleki
2023-04-24 05:58:11 +00:00
parent 073460c341
commit a9cfb82fcb
3 changed files with 105 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/bootstrap_test_cpp.cc
View File

@@ -11,7 +11,13 @@ int main()
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &worldSize);
std::shared_ptr<Bootstrap> bootstrap(new MscclppBootstrap("", rank, worldSize));
std::shared_ptr<MscclppBootstrap> bootstrap(new MscclppBootstrap(rank, worldSize));
// bootstrap->Initialize("costsim-dev-00000A:50000");
UniqueId id;
if (rank == 0)
id = bootstrap->GetUniqueId();
MPI_Bcast(&id, sizeof(id), MPI_BYTE, 0, MPI_COMM_WORLD);
bootstrap->Initialize(id);
// need to call initialization first
MPI_Finalize();