#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <mpi.h>
#include <yaxt.h>
#include "tests.h"
#include "ctest_common.h"
#include "test_redist_common.h"
static void
static void
static void
static void
int main(int argc, char **argv) {
Xt_config config = redist_exchanger_option(&argc, &argv);
MPI_Finalize();
return TEST_EXIT_CODE;
}
static void
{
enum { nvalues = 5, nselect = (nvalues + 1)/2 };
Xt_xmap xmap = build_odd_selection_xmap(nvalues, comm);
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
static const double src_data[nvalues] = {1,2,3,4,5};
double dst_data[nselect];
static const double ref_dst_data[nselect] = {1,3,5};
check_redist(redist_coll, src_data, nselect,
dst_data, fill_array_double, NULL,
ref_dst_data, MPI_DOUBLE, MPI_DOUBLE);
check_redist(redist_coll_copy, src_data, nselect,
dst_data, fill_array_double, NULL,
ref_dst_data, MPI_DOUBLE, MPI_DOUBLE);
}
static void
{
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
static const double src_data[1] = {-1};
double dst_data[1] = {-2};
static const double ref_dst_data[1] = {-2};
if (ref_dst_data[0] != dst_data[0])
PUT_ERR("error in xt_redist_s_exchange\n");
dst_data[0] = -2;
if (ref_dst_data[0] != dst_data[0])
PUT_ERR("error in xt_redist_s_exchange\n");
}
static void
{
enum { num_slice = 3,
src_slice_len = 5, dst_slice_len = (src_slice_len+1)/2 };
Xt_xmap xmap = build_odd_selection_xmap(src_slice_len, comm);
Xt_redist redists[num_slice] = {redist, redist, redist};
config);
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
{
static const double src_data[num_slice][src_slice_len]
= {{1,2,3,4,5},{6,7,8,9,10},{11,12,13,14,15}};
static const double ref_dst_data[num_slice][dst_slice_len]
= {{1,3,5},{6,8,10},{11,13,15}};
double dst_data[num_slice][dst_slice_len];
static const void *const src_data_p[num_slice]
= { src_data[0], src_data[1], src_data[2]};
void *dst_data_p[num_slice] = { dst_data[0], dst_data[1], dst_data[2] };
check_redist_coll(redist_coll, sync_mode_test_all,
num_slice, (const void **)src_data_p,
num_slice * dst_slice_len, dst_data_p, dst_data[0],
fill_array_double, NULL, ref_dst_data, MPI_DOUBLE,
MPI_DOUBLE);
}
{
static const double src_data[num_slice][src_slice_len]
= {{1,2,3,4,5},{6,7,8,9,10},{11,12,13,14,15}};
static const double ref_dst_data[num_slice][dst_slice_len]
= {{1,3,5},{6,8,10},{11,13,15}};
double dst_data[num_slice][dst_slice_len];
static const void *const src_data_p[num_slice]
= {src_data[1],src_data[0],src_data[2]};
void *dst_data_p[num_slice] = { dst_data[1], dst_data[0], dst_data[2] };
check_redist_coll(redist_coll, sync_mode_test_all,
num_slice, (const void **)src_data_p,
num_slice * dst_slice_len, dst_data_p, dst_data[0],
fill_array_double, NULL, ref_dst_data, MPI_DOUBLE,
MPI_DOUBLE);
}
{
static const double src_data[num_slice][src_slice_len]
= {{1,2,3,4,5},{6,7,8,9,10},{11,12,13,14,15}};
static const double ref_dst_data[num_slice][dst_slice_len]
= {{1,3,5},{6,8,10},{11,13,15}};
double dst_data[num_slice][3];
static const void *const src_data_p[num_slice]
= { src_data[0], src_data[1], src_data[2] };
void *dst_data_p[num_slice] = { dst_data[0], dst_data[1], dst_data[2] };
check_redist_coll(redist_coll, sync_mode_test_all,
num_slice, (const void **)src_data_p,
num_slice * dst_slice_len, dst_data_p, dst_data[0],
fill_array_double, NULL, ref_dst_data, MPI_DOUBLE,
MPI_DOUBLE);
}
}
enum { num_redists = 3 };
enum { nvalues = 5, nselect = nvalues/2+(nvalues&1) };
static void
run_displacement_check(
Xt_redist redist_coll,
int sync)
{
static const double src_data[num_redists][nvalues]
= {{1,2,3,4,5},{6,7,8,9,10},{11,12,13,14,15}};
enum { cache_size = 16, cache_overrun = 2 };
double src_data_[nvalues + cache_size + cache_overrun];
enum {
num_dst_elems = num_redists * nselect + cache_size + cache_overrun
};
double dst_data[num_dst_elems];
const void *src_data_p[num_redists] = {src_data[0],src_data[1],NULL};
void *dst_data_p[num_redists] = {dst_data+0,dst_data+nselect,NULL};
double ref_dst_data[num_dst_elems];
for (size_t j = 0; j < num_redists-1; ++j)
for (size_t i = 0; i < nselect; ++i)
ref_dst_data[j*nselect + i] = src_data[j][i*2];
for (size_t k = 0; k < cache_size + cache_overrun; ++k) {
memcpy(src_data_+k, src_data[2], nvalues * sizeof(*src_data_));
for (size_t i = 0; i < num_dst_elems; ++i)
dst_data[i] = -1;
src_data_p[2] = src_data_ + k;
dst_data_p[2] = dst_data + nselect*2 + k;
for (size_t i = 0; i < k; ++i)
ref_dst_data[(num_redists-1)*nselect + i] = -1;
for (size_t i = 0; i < nselect; ++i)
ref_dst_data[(num_redists-1)*nselect + k + i]
= src_data[num_redists-1][i*2];
for (size_t i = num_redists*nselect + k; i < num_dst_elems; ++i)
ref_dst_data[i] = -1;
check_redist_coll(redist_coll,
sync ? sync_mode_test_s : sync_mode_test_a,
num_redists, src_data_p, num_dst_elems, dst_data_p,
dst_data, fill_array_double, NULL, ref_dst_data,
MPI_DOUBLE, MPI_DOUBLE);
}
}
static void
{
Xt_xmap xmap = build_odd_selection_xmap(nvalues, comm);
Xt_redist redists[num_redists] = {redist, redist, redist};
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
run_displacement_check(redist_coll, 0);
run_displacement_check(redist_coll, 1);
run_displacement_check(redist_coll_copy, 0);
run_displacement_check(redist_coll_copy, 1);
}
add versions of standard API functions not returning on error
void xt_config_delete(Xt_config config)
void xt_initialize(MPI_Comm default_comm)
Xt_idxlist xt_idxempty_new(void)
void xt_idxlist_delete(Xt_idxlist idxlist)
void xt_redist_delete(Xt_redist redist)
Xt_redist xt_redist_copy(Xt_redist redist)
MPI_Comm xt_redist_get_MPI_Comm(Xt_redist redist)
void xt_redist_s_exchange1(Xt_redist redist, const void *src_data, void *dst_data)
Xt_redist xt_redist_collection_custom_new(Xt_redist *redists, int num_redists, int cache_size, MPI_Comm comm, Xt_config config)
Xt_redist xt_redist_p2p_new(Xt_xmap xmap, MPI_Datatype datatype)
void xt_xmap_delete(Xt_xmap xmap)
Xt_xmap xt_xmap_all2all_new(Xt_idxlist src_idxlist, Xt_idxlist dst_idxlist, MPI_Comm comm)