diff --git a/.gitignore b/.gitignore index e0d1d4e65..a60cd35c0 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,4 @@ /build* *.pyc +.vscode/ +.idea/ diff --git a/gloo/CMakeLists.txt b/gloo/CMakeLists.txt index d5e6a1a55..1092eef9d 100644 --- a/gloo/CMakeLists.txt +++ b/gloo/CMakeLists.txt @@ -43,6 +43,7 @@ list(APPEND GLOO_HDRS "${CMAKE_CURRENT_SOURCE_DIR}/broadcast_one_to_all.h" "${CMAKE_CURRENT_SOURCE_DIR}/context.h" "${CMAKE_CURRENT_SOURCE_DIR}/gather.h" + "${CMAKE_CURRENT_SOURCE_DIR}/gatherv.h" "${CMAKE_CURRENT_SOURCE_DIR}/math.h" "${CMAKE_CURRENT_SOURCE_DIR}/pairwise_exchange.h" "${CMAKE_CURRENT_SOURCE_DIR}/reduce.h" diff --git a/gloo/gatherv.cc b/gloo/gatherv.cc index 64767e5c3..67e3ea225 100644 --- a/gloo/gatherv.cc +++ b/gloo/gatherv.cc @@ -104,9 +104,7 @@ void gatherv(GathervOptions& opts) { out->waitRecv(opts.timeout); } } else { - size_t sendLength = opts.elementSize * opts.elementsPerRank[context->rank]; - GLOO_ENFORCE_GE(in->size, sendLength); - in->send(opts.root, slot, 0, sendLength); + in->send(opts.root, slot, 0, in->size); in->waitSend(opts.timeout); } }