#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <assert.h>
#include <stdbool.h>
#include <stdlib.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) {
int rank, size;
Xt_config config = redist_exchanger_option(&argc, &argv);
if (size > 1) {
{
{
int local_size[2] = {size,size};
for (size_t i = 0; i < 2; ++i)
local_start[i]);
}
{
.stride = 1, .nstrides = 2*size*size};
}
Xt_int index_vector_a[2*size*size], index_vector_b[2*size*size];
enum { dim1a = 9, rpt_cnt = 4 };
size_t dim0 = 2*(size_t)size*(size_t)size;
for (int j = 0; j < dim1a; ++j)
for (size_t i = 0; i < dim0; ++i)
input[j][i] = (
Xt_int)(index_vector_a[i] + j * 2*(
Xt_int)dim0);
static const int displacements[2][rpt_cnt] = { {0,1,2,3}, {1,2,4,8} };
MPI_Aint extent = (MPI_Aint)(dim0 *
sizeof(
Xt_int));
for (size_t i = 0; i < 2; ++i)
redist_p2p, extent, extent, rpt_cnt, displacements[i], config);
for (size_t i = 0; i < 2; ++i)
PUT_ERR("error in xt_redist_get_MPI_Comm\n");
for (int sync_mode = 0; sync_mode < 2; ++sync_mode) {
for (int j = 0; j < rpt_cnt; ++j)
for (size_t i = 0; i < dim0; ++i)
result[j][i] = -1;
for (int j = 0; j < dim1a; ++j)
for (size_t i = 0; i < dim0; ++i)
result_2[j][i] = -1;
if (sync_mode == 0) {
} else {
}
bool mismatch = false;
for (int j = 0; j < rpt_cnt; ++j)
for (size_t i = 0; i < dim0; ++i)
mismatch |= (result[j][i] != index_vector_b[i] + j * 2*(int)dim0);
if (mismatch)
PUT_ERR("ERROR: in first xt_redist_s_exchange1\n");
mismatch = false;
for (int j = 1; j <= 8; j<<=1)
for (size_t i = 0; i < dim0; ++i)
mismatch |= (result_2[j][i] != index_vector_b[i] + j * 2*(int)dim0);
if (mismatch)
PUT_ERR("ERROR: in second xt_redist_s_exchange1\n");
}
for (size_t i = 0; i < 2; ++i)
}
}
MPI_Finalize();
return TEST_EXIT_CODE;
}
void xt_config_delete(Xt_config config)
void xt_initialize(MPI_Comm default_comm)
void xt_idxlist_get_indices(Xt_idxlist idxlist, Xt_int *indices)
void xt_idxlist_delete(Xt_idxlist idxlist)
Xt_idxlist xt_idxlist_collection_new(Xt_idxlist *idxlists, int num_idxlists)
Xt_idxlist xt_idxsection_new(Xt_int start, int num_dimensions, const Xt_int global_size[num_dimensions], const int local_size[num_dimensions], const Xt_int local_start[num_dimensions])
Xt_idxlist xt_idxstripes_new(struct Xt_stripe const *stripes, int num_stripes)
#define xt_mpi_call(call, 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_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)
void xt_request_wait(Xt_request *request)
void xt_xmap_delete(Xt_xmap xmap)
Xt_xmap xt_xmap_all2all_new(Xt_idxlist src_idxlist, Xt_idxlist dst_idxlist, MPI_Comm comm)