int main (int argc, char *argv[]) {
dtype **a; /* First factor, a matrix */
dtype *b; /* Second factor, a vector */
dtype *c_block; /* Partial product vector */
dtype *c; /* Replicated product vector */
double max_seconds;
double seconds; /* Elapsed time for matrix-vector multiply */
dtype *storage; /* Matrix elements stored here */
int i, j; /* Loop indices */
int id; /* Process ID number */
int m; /* Rows in matrix */
int n; /* Columns in matrix */
int nprime; /* Elements in vector */
int p; /* Number of processes */
int rows; /* Number of rows on this process */
int its;
MPI_Init (&argc, &argv);
MPI_Comm_rank (MPI_COMM_WORLD, &id);
MPI_Comm_size (MPI_COMM_WORLD, &p);
read_row_striped_matrix (argv[1], (void *) &a,
(void *) &storage, mpitype, &m, &n, MPI_COMM_WORLD);
rows = BLOCK_SIZE(id,p,m);
print_row_striped_matrix ((void **) a, mpitype, m, n,
MPI_COMM_WORLD);
read_replicated_vector (argv[2], (void *) &b, mpitype,
&nprime, MPI_COMM_WORLD);
print_replicated_vector (b, mpitype, nprime,
MPI_COMM_WORLD);
c_block = (dtype *) malloc (rows * sizeof(dtype));
c = (dtype *) malloc (n * sizeof(dtype));
MPI_Barrier (MPI_COMM_WORLD);
seconds = - MPI_Wtime();
for (i = 0; i < rows; i++) {
c_block[i] = 0.0;
for (j = 0; j < n; j++)
c_block[i] += a[i][j] * b[j];
}
replicate_block_vector (c_block, n, (void *) c, mpitype,
MPI_COMM_WORLD);
MPI_Barrier (MPI_COMM_WORLD);
seconds += MPI_Wtime();
print_replicated_vector (c, mpitype, n, MPI_COMM_WORLD);
MPI_Allreduce (&seconds, &max_seconds, 1, mpitype, MPI_MAX,
MPI_COMM_WORLD);
if (!id) {
printf ("MV1) N = %d, Processes = %d, Time = %12.6f sec,",
n, p, max_seconds);
printf ("Mflop = %6.2f\n", 2*n*n/(1000000.0*max_seconds));
}
MPI_Finalize();
return 0;
}
int main(int argc, char * argv[]) {
dtype **a; /* First factor, a matrix */
dtype *b; /* Second factor, a vector */
dtype *c_block; /* Partial product vector */
dtype *c; /* Replicated product vector */
dtype *storage; /* Matrix elements stored here */
int i, j; /* Loop indices */
int id; /* Process ID number */
int m; /* Rows in matrix */
int n; /* Columns in matrix */
int nprime; /* Elements in vector */
int p; /* Number of processes */
int rows; /* Number of rows on this process */
MPI_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &id);
MPI_Comm_size(MPI_COMM_WORLD, &p);
read_row_striped_matrix(argv[1], (void *)&a, (void *)&storage, mpitype, &m, &n, MPI_COMM_WORLD);
rows = BLOCK_SIZE(id, p, m);
print_row_striped_matrix((void **)a, mpitype, m, n, MPI_COMM_WORLD);
read_replicated_vector(argv[2], (void *) &b, mpitype, &nprime, MPI_COMM_WORLD);
print_replicated_vector(b, mpitype, nprime, MPI_COMM_WORLD);
c_block = (dtype *)malloc(rows * sizeof(dtype));
c = (dtype *)malloc(n * sizeof(dtype));
for(i = 0; i < rows; i++) {
c_block[i] = 0.0;
for(j = 0; j < n; j++)
c_block[i] += a[i][j] * b[j];
}
replicate_block_vector(c_block, n, (void *)c, mpitype, MPI_COMM_WORLD);
print_replicated_vector(c, mpitype, n, MPI_COMM_WORLD);
MPI_Finalize();
return 0;
}