Function array_to_all

src/array_slice.cpp:48–65

/ repeatedly call sum_to_all to consolidate all entries of */ an array on all cores. */ array: in/out ptr to the data */ array_size: data size in multiples of sizeof(realnum) */ /

/* array_size: data size in multiples of sizeof(realnum)       */
/***************************************************************/
realnum *array_to_all(realnum *array, size_t array_size) {
#ifdef HAVE_MPI
  realnum *buffer = new realnum[ARRAY_TO_ALL_BUFSIZE];
  ptrdiff_t offset = 0;
  size_t remaining = array_size;
  while (remaining != 0) {
    size_t xfer_size = (remaining > ARRAY_TO_ALL_BUFSIZE ? ARRAY_TO_ALL_BUFSIZE : remaining);
    sum_to_all(array + offset, buffer, xfer_size);
    memcpy(array + offset, buffer, xfer_size * sizeof(realnum));
    remaining -= xfer_size;
    offset += xfer_size;
  }
  delete[] buffer;
#else
  UNUSED(array_size);
#endif
  return array;
}

complex<realnum> *array_to_all(complex<realnum> *array, size_t array_size) {
  return (complex<realnum> *)array_to_all((realnum *)array, 2 * array_size);