Unverified Commit 7415eaa5 authored by nv-dlasalle's avatar nv-dlasalle Committed by GitHub
Browse files

[Bugfix] Handle case where process has no elements to update, in NCCL communicator (#3035)


Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent cb2327d4
...@@ -176,7 +176,7 @@ std::pair<IdArray, NDArray> SparsePush( ...@@ -176,7 +176,7 @@ std::pair<IdArray, NDArray> SparsePush(
Workspace<DType> send_value(device, ctx, num_in*num_feat); Workspace<DType> send_value(device, ctx, num_in*num_feat);
// permute the indices and values // permute the indices and values
{ if (num_in > 0) {
const dim3 block(256); const dim3 block(256);
const dim3 grid((num_in+block.x-1)/block.x); const dim3 grid((num_in+block.x-1)/block.x);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment