diff options
Diffstat (limited to 'src/collectives/all_reduce.cc')
| -rw-r--r-- | src/collectives/all_reduce.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/collectives/all_reduce.cc b/src/collectives/all_reduce.cc index 7796d5b..b67f3be 100644 --- a/src/collectives/all_reduce.cc +++ b/src/collectives/all_reduce.cc @@ -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 ************************************************************************/ @@ -10,7 +10,8 @@ NCCL_API(ncclResult_t, ncclAllReduce, const void* sendbuff, void* recvbuff, size ncclDataType_t datatype, ncclRedOp_t op, ncclComm* comm, cudaStream_t stream); ncclResult_t ncclAllReduce(const void* sendbuff, void* recvbuff, size_t count, ncclDataType_t datatype, ncclRedOp_t op, ncclComm* comm, cudaStream_t stream) { - struct ncclInfo info = { ncclCollAllReduce, "AllReduce", + NVTX3_FUNC_RANGE_IN(nccl_domain); + struct ncclInfo info = { ncclFuncAllReduce, "AllReduce", sendbuff, recvbuff, count, datatype, op, 0, comm, stream, /* Args */ ALLREDUCE_CHUNKSTEPS, ALLREDUCE_SLICESTEPS }; return ncclEnqueueCheck(&info); |