diff options
Diffstat (limited to 'src/include/bootstrap.h')
| -rw-r--r-- | src/include/bootstrap.h | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/include/bootstrap.h b/src/include/bootstrap.h new file mode 100644 index 0000000..81af3a4 --- /dev/null +++ b/src/include/bootstrap.h @@ -0,0 +1,18 @@ +/************************************************************************* + * Copyright (c) 2015-2018, NVIDIA CORPORATION. All rights reserved. + * + * See LICENSE.txt for license information + ************************************************************************/ + +#ifndef NCCL_BOOTSTRAP_H_ +#define NCCL_BOOTSTRAP_H_ + +#include "nccl.h" + +ncclResult_t bootstrapCreateRoot(ncclUniqueId* commId, bool idFromEnv); +ncclResult_t bootstrapGetUniqueId(ncclUniqueId* out); +ncclResult_t bootstrapInit(ncclUniqueId* id, int rank, int nranks, void** commState); +ncclResult_t bootstrapAllGather(void* commState, void* allData, int size); +ncclResult_t bootstrapRingExchange(void* commState, void* prevNextData, int prev, int next, int size); +ncclResult_t bootstrapClose(void* commState); +#endif |