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 * See LICENSE.txt for license information
************************************************************************/ ************************************************************************/
@ -74,18 +74,17 @@ __global__ void ReduceKernel(const KernelArgs<T> args) {
postReadyToNext); postReadyToNext);
} else if (rank == root) { } else if (rank == root) {
Prims::Reduce( Prims::Reduce(
prevInput + boffset,
thisInput + offset, thisInput + offset,
prevInput + boffset,
thisOutput + offset, thisOutput + offset,
sliceSize, maxOffset, sliceSize, maxOffset,
step, step,
waitReadyFromPrev, waitReadyFromPrev,
postDoneToPrev); postDoneToPrev);
} else { } else {
Prims::ReduceCopy( Prims::Reduce(
thisInput + offset, thisInput + offset,
prevInput + boffset, prevInput + boffset,
thisOutput + offset,
nextOutput + boffset, nextOutput + boffset,
sliceSize, maxOffset, sliceSize, maxOffset,
step, step,