yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix crash in Reduce when non-root ranks have invalid recvbuff

Browse Source
This commit is contained in:
Sylvain Jeaugey 2017-02-22 11:55:51 -08:00
parent 024d1e2678
commit 8e1d6f9b60
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/reduce.cu
View File

@ -1,5 +1,5 @@
/*************************************************************************
* Copyright (c) 2015-2016, NVIDIA CORPORATION. All rights reserved.
* Copyright (c) 2015-2017, NVIDIA CORPORATION. All rights reserved.
*
* See LICENSE.txt for license information
************************************************************************/
@ -74,18 +74,17 @@ __global__ void ReduceKernel(const KernelArgs<T> args) {
postReadyToNext);
} else if (rank == root) {
Prims::Reduce(
prevInput + boffset,
thisInput + offset,
prevInput + boffset,
thisOutput + offset,
sliceSize, maxOffset,
step,
waitReadyFromPrev,
postDoneToPrev);
} else {
Prims::ReduceCopy(
Prims::Reduce(
thisInput + offset,
prevInput + boffset,
thisOutput + offset,
nextOutput + boffset,
sliceSize, maxOffset,
step,