We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a2a23b0 commit 4ca10e4Copy full SHA for 4ca10e4
tests_nccl/test_ncclutils_nccl.py
@@ -83,7 +83,7 @@ def test_allgather_differentsize_withrecbuf(par):
83
84
# Gathered array
85
send_shapes = MPI.COMM_WORLD.allgather(local_array.shape)
86
- (send_buf, recv_buf) = _prepare_nccl_allgather_inputs(local_array, send_shapes)
+ send_buf, recv_buf = _prepare_nccl_allgather_inputs(local_array, send_shapes)
87
recv_buf = nccl_allgather(nccl_comm, send_buf, recv_buf)
88
chunks = _unroll_nccl_allgather_recv(recv_buf, send_buf.shape, send_shapes)
89
gathered_array = cp.concatenate(chunks)
0 commit comments