1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include "mpitestconf.h"
#include "mpi.h"
#ifdef HAVE_IOSTREAM
// Not all C++ compilers have iostream instead of iostream.h
#include <iostream>
#ifdef HAVE_NAMESPACE_STD
// Those that do often need the std namespace; otherwise, a bare "cout"
// is likely to fail to compile
using namespace std;
#endif
#else
#include <iostream.h>
#endif
#ifdef HAVE_STRING_H
#include <string.h>
#endif
#include "mpitestcxx.h"
static int ncalls = 0;
void efn(MPI::Comm & comm, int *code, ...)
{
ncalls++;
}
int main(int argc, char *argv[])
{
MPI::Errhandler eh;
int size;
bool foundMsg;
int errs = 0;
MTest_Init();
size = MPI::COMM_WORLD.Get_size();
// Test that we can change the default error handler to not throw
// an exception (our error handler could also throw an exception,
// but we want to make sure that the exception is not thrown
// by the MPI code)
eh = MPI::Comm::Create_errhandler(efn);
MPI::COMM_WORLD.Set_errhandler(eh);
try {
foundMsg = MPI::COMM_WORLD.Iprobe(size, 0);
} catch(MPI::Exception ex) {
cout << "Caught exception from iprobe (should have called error handler instead)\n";
errs++;
if (ex.Get_error_class() == MPI_SUCCESS) {
errs++;
cout << "Unexpected error from Iprobe" << endl;
}
}
if (ncalls != 1) {
errs++;
cout << "Did not invoke error handler when invoking iprobe with an invalid rank" << endl;
}
eh.Free();
// Find out how many errors we saw
MTest_Finalize(errs);
return 0;
}
|