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
72
73
74
75
76
77
78
79
80
|
/***************************************************************************/
/* ____Demonstrates use of shared and non-shared streams____ */
/* Each process has two streams. One stream is common to all the */
/* processes. The other stream is different on each process. */
/***************************************************************************/
#include <stdio.h>
#include <mpi.h> /* MPI header file */
/* Uncomment the following line to get the interface with pointer checking */
/*#define CHECK_POINTERS */
#include "sprng.h" /* SPRNG header file */
#define SEED 985456376
main(int argc, char *argv[])
{
int streamnum, commNum, nstreams, *stream, *commonStream;
double rn;
int i, myid, nprocs;
int gtype; /*--- */
/************************** MPI calls ***********************************/
MPI_Init(&argc, &argv); /* Initialize MPI */
MPI_Comm_rank(MPI_COMM_WORLD, &myid); /* find process id */
MPI_Comm_size(MPI_COMM_WORLD, &nprocs); /* find number of processes */
/****************** Initialization values *******************************/
streamnum = myid; /*This stream is different on each process*/
commNum = nprocs; /* This stream is common to all processes */
nstreams = nprocs + 1; /* extra stream is common to all processes*/
/*********************** Initialize streams *****************************/
/*--- node 0 is reading in a generator type */
if(myid == 0)
{
#include "gen_types_menu.h"
printf("Type in a generator type (integers: 0,1,2,3,4,5): ");
scanf("%d", >ype);
}
MPI_Bcast(>ype,1,MPI_INT,0,MPI_COMM_WORLD ); /*--- broadcast gen type */
/* This stream is different on each process */
stream = init_sprng(gtype,streamnum,nstreams,SEED,SPRNG_DEFAULT);
printf("Process %d: Print information about new stream\n", myid);
print_sprng(stream);
/* This stream is identical on each process */
commonStream = init_sprng(gtype,commNum,nstreams,SEED,SPRNG_DEFAULT);
printf("Process %d: This stream is identical on all processes\n", myid);
print_sprng(commonStream);
/*********************** print random numbers ***************************/
for (i=0;i<2;i++) /* random numbers from distinct stream */
{
rn = sprng(stream); /* generate double precision random number*/
printf("Process %d, random number (distinct stream) %d: %f\n",
myid, i+1, rn);
}
for (i=0;i<2;i++) /* random number from common stream */
{
rn = sprng(commonStream); /*generate double precision random number */
printf("Process %d, random number (shared stream) %d: %f\n", myid, i+1, rn);
}
/*************************** free memory ********************************/
free_sprng(stream); /* free memory used to store stream state */
free_sprng(commonStream);
MPI_Finalize(); /* terminate MPI */
}
|
