#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdbool.h>
#include <string.h>
#include <mpi.h>
#include <yaxt.h>
#include "tests.h"
#include "ctest_common.h"
#include "test_redist_common.h"
int main(int argc, char **argv) {
Xt_config config = redist_exchanger_option(&argc, &argv);
{
enum { src_slice_len = 5, dst_slice_len = 3 };
Xt_xmap xmap = build_odd_selection_xmap(src_slice_len, comm);
static const double src_data[] = {1,2,3,4,5};
double dst_data[dst_slice_len];
MPI_Aint src_extent = (MPI_Aint)(sizeof(src_data)),
dst_extent = (MPI_Aint)(sizeof(dst_data));
static const int displacements[1] = {0};
redist, src_extent, dst_extent, 1, displacements, config);
comm))
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
static const double ref_dst_data[] = {1,3,5};
check_redist(redist_repeat, src_data, dst_slice_len, dst_data,
fill_array_double, NULL, ref_dst_data, MPI_DOUBLE, MPI_DOUBLE);
}
{
enum { src_slice_len = 5, dst_slice_len = (src_slice_len+1)/2 };
Xt_xmap xmap = build_odd_selection_xmap(src_slice_len, comm);
enum { num_repetitions = 3 };
static const double src_data[num_repetitions][src_slice_len]
= {{1,2,3,4,5},{6,7,8,9,10},{11,12,13,14,15}};
double dst_data[num_repetitions][dst_slice_len];
MPI_Aint src_extent =
(MPI_Aint)((size_t)(src_data[1] - src_data[0]) * sizeof (double));
MPI_Aint dst_extent =
(MPI_Aint)((size_t)(dst_data[1] - dst_data[0]) * sizeof (double));
static const int displacements[num_repetitions] = {0, 1, 2};
redist, src_extent, dst_extent,
num_repetitions, displacements, config);
comm))
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
static const double ref_dst_data[num_repetitions][dst_slice_len]
= {{1,3,5},{6,8,10},{11,13,15}};
check_redist(redist_repeat, src_data, num_repetitions * dst_slice_len,
dst_data, fill_array_double, NULL, ref_dst_data,
MPI_DOUBLE, MPI_DOUBLE);
}
{
enum { src_slice_len = 5, dst_slice_len = (src_slice_len+1)/2 };
Xt_xmap xmap = build_odd_selection_xmap(src_slice_len, comm);
enum { num_repetitions = 3 };
static const double src_data[num_repetitions][src_slice_len]
= {{1,2,3,4,5},{6,7,8,9,10},{11,12,13,14,15}};
double dst_data[num_repetitions][dst_slice_len];
MPI_Aint src_extent =
(MPI_Aint)((size_t)(src_data[1] - src_data[0]) * sizeof (double));
MPI_Aint dst_extent =
(MPI_Aint)((size_t)(dst_data[1] - dst_data[0]) * sizeof (double));
static const int src_displacements[num_repetitions] = {0, 1, 2},
dst_displacements[num_repetitions] = {2, 1, 0};
num_repetitions, src_displacements,
dst_displacements, config);
comm))
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
static const double ref_dst_data[num_repetitions][dst_slice_len]
= {{11,13,15},{6,8,10},{1,3,5}};
check_redist(redist_repeat, src_data, num_repetitions * dst_slice_len,
dst_data, fill_array_double, NULL, ref_dst_data,
MPI_DOUBLE, MPI_DOUBLE);
}
{
enum { src_slice_len = 5, dst_slice_len = (src_slice_len+1)/2,
rep_dim_size = 5 };
Xt_xmap xmap = build_odd_selection_xmap(src_slice_len, comm);
MPI_Aint src_extent = (MPI_Aint)(src_slice_len * sizeof (double));
MPI_Aint dst_extent = (MPI_Aint)(dst_slice_len * sizeof (double));
static const int displacements[3] = {0, 2, 4};
redist, src_extent, dst_extent, 3, displacements, config);
comm))
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
static const double src_data[rep_dim_size][src_slice_len]
= {{1,2,3,4,5},{-1},{6,7,8,9,10},{-1},{11,12,13,14,15}};
double dst_data[rep_dim_size][dst_slice_len];
static const double ref_dst_data[rep_dim_size][dst_slice_len]
= {{1,3,5},{-1,-1,-1},{6,8,10},{-1,-1,-1},{11,13,15}};
check_redist(redist_repeat, src_data, rep_dim_size * dst_slice_len,
dst_data, fill_array_double, NULL, ref_dst_data,
MPI_DOUBLE, MPI_DOUBLE);
}
{
enum { npt = 9 };
Xt_xmap xmap = build_odd_selection_xmap(6, comm);
static const int src_pos[npt] = {1,2,3,4,5,6,7,8,9};
static const int dst_pos[npt] = {0,2,4,6,8,10,12,14,16};
xmap, src_pos, dst_pos, MPI_DOUBLE, config);
double src_data[npt], dst_data[npt];
static const double ref_dst_data[npt]
= {102.0, 103.0, 104.0, 105.0, 106.0, 107.0, -1.0, -1.0, -1.0};
for(int i = 0; i<npt; i++) src_data[i] = 100+i+1;
for (int j = 0; j < 2; ++j) {
for(int i = 0; i<npt; i++) dst_data[i] = -1;
if (j == 0) {
} else {
}
bool mismatch = false;
for(int i=0; i<npt; i++)
mismatch |= (dst_data[i] != ref_dst_data[i]);
if (mismatch)
PUT_ERR("error in xt_redist_repeat\n");
}
MPI_Aint src_extent = (MPI_Aint)(1 * sizeof (double));
MPI_Aint dst_extent = (MPI_Aint)(1 * sizeof (double));
static const int displacements[2] = {0,1};
redist, src_extent, dst_extent, 2, displacements, config);
comm))
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
check_redist(redist_repeat, src_data, npt, dst_data,
fill_array_double, NULL, ref_dst_data, MPI_DOUBLE, MPI_DOUBLE);
}
MPI_Finalize();
return TEST_EXIT_CODE;
}
void xt_config_delete(Xt_config config)
void xt_initialize(MPI_Comm default_comm)
void xt_redist_delete(Xt_redist redist)
void xt_redist_a_exchange1(Xt_redist redist, const void *src_data, void *dst_data, Xt_request *request)
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_p2p_new(Xt_xmap xmap, MPI_Datatype datatype)
Xt_redist xt_redist_p2p_off_custom_new(Xt_xmap xmap, const int *src_offsets, const int *dst_offsets, MPI_Datatype datatype, Xt_config config)
Xt_redist xt_redist_repeat_custom_new(Xt_redist redist, MPI_Aint src_extent, MPI_Aint dst_extent, int num_repetitions, const int displacements[num_repetitions], Xt_config config)
Xt_redist xt_redist_repeat_asym_custom_new(Xt_redist redist, MPI_Aint src_extent, MPI_Aint dst_extent, int num_repetitions, const int src_displacements[num_repetitions], const int dst_displacements[num_repetitions], Xt_config config)
void xt_request_wait(Xt_request *request)
void xt_xmap_delete(Xt_xmap xmap)