diff options
Diffstat (limited to 'src/include/trees.h')
| -rw-r--r-- | src/include/trees.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/include/trees.h b/src/include/trees.h index 7eadd85..ded84a6 100644 --- a/src/include/trees.h +++ b/src/include/trees.h @@ -1,5 +1,5 @@ /************************************************************************* - * Copyright (c) 2015-2019, NVIDIA CORPORATION. All rights reserved. + * Copyright (c) 2015-2020, NVIDIA CORPORATION. All rights reserved. * * See LICENSE.txt for license information ************************************************************************/ @@ -7,7 +7,7 @@ #ifndef NCCL_TREES_H_ #define NCCL_TREES_H_ -ncclResult_t ncclGetBtree(int nranks, int rank, int* u0, int* d1, int* d0); -ncclResult_t ncclGetDtree(int nranks, int rank, int* u0, int* d0_0, int* d0_1, int* u1, int* d1_0, int* d1_1); +ncclResult_t ncclGetBtree(int nranks, int rank, int* u0, int* d1, int* d0, int* parentChildType); +ncclResult_t ncclGetDtree(int nranks, int rank, int* u0, int* d0_0, int* d0_1, int* parentChildType0, int* u1, int* d1_0, int* d1_1, int* parentChildType1); #endif |