SHMEM_ALLTOALL32(3)

πŸ¦‹ NAME πŸ¦‹

okay so bestie, listen up β€” shmem_alltoall32(3), shmem_alltoall64(3), shmem_alltoalls32(3), and shmem_alltoalls64(3) are the collective routines where every PE exchanges a fixed amount of data with literally every other PE in the Active set. it's giving group project where everyone actually participates. πŸ’…βœ¨


✨ SYNOPSIS ✨

🌸 C or C++:

#include <mpp/shmem.h>

void shmem_alltoall32(void *target, const void *source,
  size_t nelems, int PE_start, int logPE_stride,
  int PE_size, long *pSync)

void shmem_alltoall64(void *target, const void *source,
  size_t nelems, int PE_start, int logPE_stride,
  int PE_size, long *pSync)

void shmem_alltoalls32(void *target, const void *source,
  ptrdiff_t dst, ptrdiff_t sst,
  size_t nelems, int PE_start, int logPE_stride,
  int PE_size, long *pSync)

void shmem_alltoalls64(void *target, const void *source,
  ptrdiff_t dst, ptrdiff_t sst,
  size_t nelems, int PE_start, int logPE_stride,
  int PE_size, long *pSync)

πŸŽ€ Fortran:

INCLUDE "mpp/shmem.fh"

INTEGER dst, sst, nelems, PE_root, PE_start, logPE_stride, PE_size
INTEGER pSync(SHMEM_ALLTOALL_SYNC_SIZE)

CALL SHMEM_ALLTOALL32(target, source, nelems,
& PE_start, logPE_stride, PE_size, fIpSync)

CALL SHMEM_ALLTOALL64(target, source, nelems,
& PE_start, logPE_stride, PE_size, pSync)

CALL SHMEM_ALLTOALLS32(target, source, dst, sst, nelems,
& PE_start, logPE_stride, PE_size, pSync)

CALL SHMEM_ALLTOALLS64(target, source, dst, sst, nelems,
& PE_start, logPE_stride, PE_size, pSync)

πŸ’– DESCRIPTION πŸ’–

mkay so the shmem_alltoalls routines are collective routines. Each PE in the Active set exchanges nelems strided data elements β€” sized 32 bits (for shmem_alltoalls32) or 64 bits (for shmem_alltoalls64) β€” with all the other PEs in the set. everybody gets a lil something from everybody, it's so cute 🫢

Both strides, dst and sst, must be greater than or equal to 1. no zero-striding, we don't do that here.

Here's the mapping, bestie, write it down: the sst*jth block sent from PE i to PE j lands in the dst*ith block of the dest data object on PE j.

As with all OpenSHMEM collective routines, these routines assume that only PEs in the Active set call them. If a PE that is not in the Active set calls an OpenSHMEM collective routine β†’ undefined behavior results. don't do it. for real. 😭

The values of the arguments dst, sst, nelems, PE_start, logPE_stride, and PE_size must be equal on all PEs in the Active set. And the same dest and source data objects, plus the same pSync work array, must be passed to all PEs in the Active set. everyone on the same page, no rogue vibes.

Before any PE calls a shmem_alltoalls routine, these conditions must exist (synchronization via a barrier or some other method is often needed to make sure of it):

  • The pSync array on all PEs in the Active set is not still in use from a prior call to a shmem_alltoalls routine.
  • The dest data object on all PEs in the Active set is ready to accept the shmem_alltoalls data.

Upon return from a shmem_alltoalls routine, the following is true for the local PE:

  • Its dest symmetric data object is completely updated, and the data has been copied out of the source data object.
  • The values in the pSync array are restored to the original values.

🧚 the arguments 🧚

The arguments are as follows:

First, the symmetric data object needs to be one of these data types:

  • shmem_alltoall32: Any noncharacter type that has an element size of 32 bits. No Fortran derived types or C/C++ structures are allowed.
  • shmem_alltoall64: Any noncharacter type that has an element size of 64 bits. No Fortran derived types or C/C++ structures are allowed.

and then, the actual args:

  • target β€” a symmetric data object big enough to receive the combined total of nelems elements from every PE in the Active set. size it right, bestie, no shortcuts.
  • source β€” a symmetric data object containing nelems elements of data for each PE in the Active set, ordered according to destination PE.
  • dst β€” the stride between consecutive elements of the dest data object. The stride is scaled by the element size. A value of 1 means contiguous data. dst must be of type ptrdiff_t. In Fortran it must be a default integer value.
  • sst β€” the stride between consecutive elements of the source data object. The stride is scaled by the element size. A value of 1 means contiguous data. sst must be of type ptrdiff_t. In Fortran it must be a default integer value.
  • nelems β€” the number of elements to exchange for each PE. nelems must be of type size_t for C/C++. In Fortran it must be a default integer value.
  • PE_start β€” the lowest virtual PE number of the active set of PEs. PE_start must be of type integer. In Fortran it must be a default integer value.
  • logPE_stride β€” the log (base 2) of the stride between consecutive virtual PE numbers in the active set. log_PE_stride must be of type integer. In Fortran it must be a default integer value.
  • PE_size β€” the number of PEs in the active set. PE_size must be of type integer. In Fortran it must be a default integer value.
  • pSync β€” a symmetric work array. In C/C++, pSync must be of type long and size _SHMEM_ALLTOALL_SYNC_SIZE. In Fortran, pSync must be of type integer and size SHMEM_ALLTOALL_SYNC_SIZE. Every element of this array must be initialized with the value _SHMEM_SYNC_VALUE (in C/C++) or SHMEM_SYNC_VALUE (in Fortran) before any of the PEs in the active set enter shmem_barrier(). πŸ”₯

Upon return from a shmem_alltoalls routine, the following is true for the local PE: Its dest symmetric data object is completely updated and the data has been copied out of the source data object. The values in the pSync array are restored to the original values.

The values of arguments PE_root, PE_start, logPE_stride, and PE_size must be equal on all PEs in the active set. The same target and source data objects and the same pSync work array must be passed to all PEs in the active set.

Before any PE calls a alltoall routine, you must ensure that the following conditions exist (synchronization via a barrier or some other method is often needed to ensure this): The pSync array on all PEs in the active set is not still in use from a prior call to a alltoall routine. The target array on all PEs in the active set is ready to accept the alltoall data.


πŸ“ NOTES πŸ“

The terms collective and symmetric are defined in intro_shmem(3). go read it, iykyk πŸ’œ

All SHMEM alltoall routines restore pSync to its original contents. Multiple calls to SHMEM routines that use the same pSync array do not require that pSync be reinitialized after the first call. slay, one less chore.

You must ensure the that the pSync array is not being updated by any PE in the active set while any of the PEs participates in processing of a SHMEM broadcast routine. Be careful to avoid these situations: If the pSync array is initialized at run time, some type of synchronization is needed to ensure that all PEs in the working set have initialized pSync before any of them enter a SHMEM routine called with the pSync synchronization array. A pSync array may be reused on a subsequent SHMEM broadcast routine only if none of the PEs in the active set are still processing a prior SHMEM alltoall routine call that used the same pSync array. In general, this can be ensured only by doing some type of synchronization. However, in the special case of SHMEM routines being called with the same active set, you can allocate two pSync arrays and alternate between them on successive calls. 🧠✨

fr fr though β€” that pSync reuse rule is the number one way people break their code. double-buffer those arrays if you have to.

🌈 EXAMPLES 🌈

πŸ’œ C/C++ example:

#include <shmem.h>
#include <stdio.h>

long pSync[SHMEM_ALLTOALL_SYNC_SIZE];
int main(void)
{
int64_t *source, *dest;
int i, count, pe;
shmem_init();
count = 2;
dest = (int64_t*) shmem_malloc(count * shmem_n_pes() * sizeof(int64_t));
source = (int64_t*) shmem_malloc(count * shmem_n_pes() * sizeof(int64_t));
/* assign source values */
for (pe=0; pe <shmem_n_pes(); pe++){
for (i=0; i<count; i++){
source[(pe*count)+i] = shmem_my_pe() + pe;
dest[(pe*count)+i] = 9999;
}
}
for (i=0; i< SHMEM_ALLTOALLS_SYNC_SIZE; i++) {
pSync[i] = SHMEM_SYNC_VALUE;
}
/* wait for all PEs to initialize pSync */
shmem_barrier_all();
/* alltoalls on all PES */
shmem_alltoalls64(dest, source, 1, 1, count, 0, 0, shmem_n_pes(), pSync);
/* verify results */
for (pe=0; pe<shmem_n_pes(); pe++) {
for (i=0; i<count; i++){
if (dest[(pe*count)+i] != shmem_my_pe() + pe) {
printf("[%d] ERROR: dest[%d]=%ld, should be %d,
shmem_my_pe(),(pe*count)+i,dest[(pe*count)+i],
shmem_n_pes() + pe);
}
}
}
shmem_barrier_all();
shmem_free(dest);
shmem_free(source);
shmem_finalize();
return 0;
}

See also: intro_shmem(3)


2003-2026, The Open MPI Community

okay that's the whole page, you ate it, go write some correct collectives πŸ’–βœ¨ and hey β€” this gorgeous lil manual is from Open MPI, dated Sep 10, 2026, and it's SHMEM_ALLTOALL32(3). mwah. πŸŽ€πŸ¦‹ no cap, you've got this. πŸ‘‘

πŸ“¦ where this page comes from