Fix incorrect pointer checking for scatter and gather (#285)

[ROCm/rccl commit: dcad0ef7cb]
这个提交包含在:
Wenkai Du 编写, GitHub 于 2020-10-19 13:27:09 -07:00 提交
1 父节点 f4b9a0d8e5
当前提交 1aae6b1344
1 个文件已更改
+2 -2
+2 -2
查看文件
@@ -73,10 +73,10 @@ ncclResult_t ArgsCheck(struct ncclInfo* info) {
}
} else {
// Check CUDA device pointers
if (info->coll != ncclCollBroadcast || info->comm->rank == info->root) {
if ((info->coll != ncclCollBroadcast && info->coll != ncclCollScatter) || info->comm->rank == info->root) {
NCCLCHECK(CudaPtrCheck(info->sendbuff, info->comm, "sendbuff", info->opName));
}
if (info->coll != ncclCollReduce || info->comm->rank == info->root) {
if ((info->coll != ncclCollReduce && info->coll != ncclCollGather) || info->comm->rank == info->root) {
NCCLCHECK(CudaPtrCheck(info->recvbuff, info->comm, "recvbuff", info->opName));
}
}