Fix typo calling allreduce with the allgather module.
That was causing CUDA collective to crash.
Этот коммит содержится в:
родитель
394e23d179
Коммит
61e900eea5
@ -811,7 +811,7 @@ int ompi_comm_split_type (ompi_communicator_t *comm, int split_type, int key,
|
|||||||
tmp[3] = -key;
|
tmp[3] = -key;
|
||||||
|
|
||||||
rc = comm->c_coll.coll_allreduce (MPI_IN_PLACE, &tmp, 4, MPI_INT, MPI_MAX, comm,
|
rc = comm->c_coll.coll_allreduce (MPI_IN_PLACE, &tmp, 4, MPI_INT, MPI_MAX, comm,
|
||||||
comm->c_coll.coll_allgather_module);
|
comm->c_coll.coll_allreduce_module);
|
||||||
if (OPAL_UNLIKELY(OMPI_SUCCESS != rc)) {
|
if (OPAL_UNLIKELY(OMPI_SUCCESS != rc)) {
|
||||||
return rc;
|
return rc;
|
||||||
}
|
}
|
||||||
|
Загрузка…
x
Ссылка в новой задаче
Block a user