Skip to content

Commit 40f7aae

Browse files
authored
Merge branch 'kokkos:develop' into simple-init
2 parents de6c8b9 + f1ad0de commit 40f7aae

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/impl/KokkosComm_alltoall.hpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,9 @@ void alltoall(const ExecSpace &space, const SendView &sv, const size_t sendCount
5555
static_assert(KokkosComm::rank<SendView>() <= 1, "alltoall for SendView::rank > 1 not supported");
5656
static_assert(KokkosComm::rank<RecvView>() <= 1, "alltoall for RecvView::rank > 1 not supported");
5757

58+
// Make sure views are ready
59+
space.fence("KokkosComm::Impl::alltoall");
60+
5861
if (KokkosComm::PackTraits<SendView>::needs_pack(sv) || KokkosComm::PackTraits<RecvView>::needs_pack(rv)) {
5962
throw std::runtime_error("alltoall for non-contiguous views not implemented");
6063
} else {
@@ -90,6 +93,9 @@ void alltoall(const ExecSpace &space, const RecvView &rv, const size_t recvCount
9093

9194
static_assert(RecvView::rank <= 1, "alltoall for RecvView::rank > 1 not supported");
9295

96+
// Make sure views are ready
97+
space.fence("KokkosComm::Impl::alltoall");
98+
9399
if (KokkosComm::PackTraits<RecvView>::needs_pack(rv)) {
94100
throw std::runtime_error("alltoall for non-contiguous views not implemented");
95101
} else {

0 commit comments

Comments
 (0)