Skip to content

Commit

Permalink
test: add basic large API tests
Browse files Browse the repository at this point in the history
  • Loading branch information
hzhou committed Feb 18, 2021
1 parent 234db6d commit d1a3b03
Show file tree
Hide file tree
Showing 9 changed files with 283 additions and 0 deletions.
1 change: 1 addition & 0 deletions test/mpi/coll/Makefile.am
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,7 @@ noinst_PROGRAMS = \
neighb_alltoallw \
bcast \
bcast_comm_world_only \
coll_large \
ring_neighbor_alltoall

allgatherv4_LDADD = $(LDADD) -lm
Expand Down
67 changes: 67 additions & 0 deletions test/mpi/coll/coll_large.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,67 @@
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/

#include "mpi.h"
#include "mpitest.h"
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

/*
Basic test for collective large APIs.
It is only intended to test the API interface rather than stress testing the internals.
*/

#define MAX_SIZE 1024
#define MAX_PROC 20

MPI_Comm comm;
int rank, nprocs;
int errs;
int sendbuf[MAX_SIZE];
int recvbuf[MAX_SIZE];

static void init_buf(void)
{
for (int i = 0; i < MAX_SIZE; i++) {
sendbuf[i] = i;
}
for (int i = 0; i < MAX_SIZE; i++) {
recvbuf[i] = -1;
}
}

static void test_alltoallv(void)
{
MPI_Count counts[MAX_PROC];
MPI_Aint displs[MAX_PROC];
for (int i = 0; i < nprocs; i++) {
counts[i] = 1;
}
for (int i = 0; i < nprocs; i++) {
displs[i] = i * 3;
}
init_buf();
MPI_Alltoallv_c(sendbuf, counts, displs, MPI_INT, recvbuf, counts, displs, MPI_INT, comm);
/* check results */
for (int i = 0; i < nprocs; i++) {
if (recvbuf[displs[i]] != rank * 3) {
errs++;
}
}
}

int main(int argc, char **argv)
{
MTest_Init(&argc, &argv);
comm = MPI_COMM_WORLD;
MPI_Comm_size(comm, &nprocs);
MPI_Comm_rank(comm, &rank);

test_alltoallv();

MTest_Finalize(errs);
return MTestReturnValue(errs);
}
2 changes: 2 additions & 0 deletions test/mpi/coll/testlist.in
Original file line number Diff line number Diff line change
Expand Up @@ -174,3 +174,5 @@ ring_neighbor_alltoall 1
# Tests for persistent collectives
p_bcast 4 strict=false
p_bcast2 8 strict=false

coll_large 8
1 change: 1 addition & 0 deletions test/mpi/datatype/Makefile.am
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ noinst_PROGRAMS = \
large_type \
large_type_sendrec \
large_vec \
type_large \
lbub \
lbub_oldapi \
localpack \
Expand Down
1 change: 1 addition & 0 deletions test/mpi/datatype/testlist.in
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,7 @@ vecblklen 1
hvecblklen 1
longdouble 1
dataalign 2
type_large 1
@largetest@large_count 1
cxx_types 1
@largetest@large_type 1
Expand Down
138 changes: 138 additions & 0 deletions test/mpi/datatype/type_large.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,138 @@
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/

#include <stdio.h>
#include <stdlib.h>
#include <mpi.h>
#include "mpitest.h"

/* Basic test to test MPI 4.0 large API */

static MPI_Datatype make_large_contig(MPI_Count nbytes)
{
MPI_Datatype newtype;
MPI_Type_contiguous_c(nbytes, MPI_CHAR, &newtype);
MPI_Type_commit(&newtype);
return newtype;
}

static MPI_Datatype make_large_indexed_block(MPI_Count nbytes)
{
MPI_Datatype newtype;

#undef NBLK
#define NBLK 256
MPI_Count num_blks = NBLK;
MPI_Count blklen = nbytes / num_blks / sizeof(int);
MPI_Count displs[NBLK];

if (num_blks * blklen * sizeof(int) != nbytes) {
return MPI_DATATYPE_NULL;
}

for (int i = 0; i < NBLK; i++) {
displs[i] = i * blklen;
}

MPI_Type_create_indexed_block_c(NBLK, blklen, displs, MPI_INT, &newtype);
MPI_Type_commit(&newtype);

return newtype;
}

static MPI_Datatype make_large_hindexed(MPI_Count nbytes)
{
MPI_Datatype newtype;

#undef NBLK
#define NBLK 4
MPI_Count blkls[NBLK];
MPI_Count displs[NBLK];

MPI_Count pos = 0;
for (int i = 0; i < NBLK - 1; i++) {
blkls[i] = i;
displs[i] = pos;
pos += blkls[i];
}
blkls[NBLK - 1] = nbytes - pos;
displs[NBLK - 1] = pos;

MPI_Type_create_hindexed_c(NBLK, blkls, displs, MPI_CHAR, &newtype);
MPI_Type_commit(&newtype);

return newtype;
}


int testtype(MPI_Datatype type, MPI_Count expected)
{
MPI_Count size, lb, extent;
int errs = 0;
MPI_Type_size_c(type, &size);

if (size < 0) {
printf("ERROR: type size apparently overflowed integer\n");
errs++;
}

if (size != expected) {
printf("reported type size %lld does not match expected %lld\n", (long long) size,
(long long) expected);
errs++;
}

MPI_Type_get_true_extent_c(type, &lb, &extent);
if (lb != 0) {
printf("ERROR: type should have lb of 0, reported %lld\n", (long long) lb);
errs++;
}

if (extent != size) {
printf("ERROR: extent should match size, not %lld\n", (long long) extent);
errs++;
}
return errs;
}


int main(int argc, char **argv)
{

int errs = 0, i;
int rank, size;
MTest_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &size);

#define NR_TYPES 3
MPI_Offset expected_sizes[NR_TYPES] = { 1024UL * 1024UL * 2400UL * 2,
2346319872 * 3,
2346319872 * 4
};
MPI_Datatype types[NR_TYPES];

if (sizeof(MPI_Aint) == 4) {
/* 32-bit system can't work with sizes more than address int */
for (int i = 0; i < NR_TYPES; i++) {
expected_sizes[i] /= 8;
}
}

types[0] = make_large_contig(expected_sizes[0]);
types[1] = make_large_indexed_block(expected_sizes[1]);
types[2] = make_large_hindexed(expected_sizes[2]);

for (i = 0; i < NR_TYPES; i++) {
if (types[i] != MPI_DATATYPE_NULL) {
errs += testtype(types[i], expected_sizes[i]);
MPI_Type_free(&(types[i]));
}
}

MTest_Finalize(errs);

return MTestReturnValue(errs);
}
1 change: 1 addition & 0 deletions test/mpi/pt2pt/Makefile.am
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,7 @@ noinst_PROGRAMS = \
pingping_barrier \
sendrecv1 \
sendself \
pt2pt_large \
precv_anysrc \
precv_anysrc_exp

Expand Down
71 changes: 71 additions & 0 deletions test/mpi/pt2pt/pt2pt_large.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/

#include "mpi.h"
#include <stdlib.h>
#include <stdio.h>
#include "mpitest.h"

/* Basic tests for large pt2pt APIs. It does not actually send large data */

#define SIZE 1024

int rank, nproc;
int errs;
int sendbuf[SIZE];
int recvbuf[SIZE];

static void init_buf(void)
{
if (rank == 0) {
for (int i = 0; i < SIZE; i++) {
sendbuf[i] = i;
}
} else if (rank == 1) {
for (int i = 0; i < SIZE; i++) {
recvbuf[i] = -1;
}
}
}

static void check_buf(void)
{
if (rank == 1) {
for (int i = 0; i < SIZE; i++) {
if (recvbuf[i] != i) {
errs++;
}
}
}
}


int main(int argc, char *argv[])
{
MPI_Comm comm;
MPI_Count count = SIZE;
int tag = 0;

MTest_Init(&argc, &argv);

comm = MPI_COMM_WORLD;
MPI_Comm_rank(comm, &rank);
MPI_Comm_size(comm, &nproc);
if (nproc != 2) {
printf("Run test with 2 procs!\n");
return 1;
}

init_buf();
if (rank == 0) {
MPI_Send_c(sendbuf, count, MPI_INT, 1, tag, comm);
} else if (rank == 1) {
MPI_Recv_c(recvbuf, count, MPI_INT, 0, tag, comm, MPI_STATUS_IGNORE);
}
check_buf();

MTest_Finalize(errs);
return MTestReturnValue(errs);
}
1 change: 1 addition & 0 deletions test/mpi/pt2pt/testlist.in
Original file line number Diff line number Diff line change
Expand Up @@ -63,3 +63,4 @@ large_tag 2
precv_anysrc 2 timeLimit=60
precv_anysrc 2 timeLimit=60 env=MPIR_CVAR_CH4_OFI_AM_LONG_FORCE_PIPELINE=true
precv_anysrc_exp 2 timeLimit=60
pt2pt_large 2

0 comments on commit d1a3b03

Please sign in to comment.