LBPM/tests/TestSetDevice.cpp

#include <iostream>
#include "common/MPI_Helpers.h"
#include "common/Utilities.h"
#include "common/ScaLBL.h"

int main (int argc, char **argv)
{
	MPI_Init(&argc,&argv);
    int rank = MPI_WORLD_RANK();
    int nprocs = MPI_WORLD_SIZE();

    for (int i=0; i<nprocs; i++) {
        if ( rank==i )
            printf("%i of %i: Hello world\n",rank,nprocs);
        MPI_Barrier(MPI_COMM_WORLD);
    }

    // Initialize compute device
    ScaLBL_SetDevice(rank);
    ScaLBL_DeviceBarrier();
    MPI_Barrier(MPI_COMM_WORLD);

    // Create a memory leak for valgrind to find
    if ( nprocs==1 ) {
        double *x = new double[1];
        ASSERT(x!=NULL);
    }

    // set the error code
    // Note: the error code should be consistent across all processors
    int error = 0;
    
    // Finished
	MPI_Barrier(MPI_COMM_WORLD);
	MPI_Finalize();
    return error; 
}