C++ (Cpp) print_row_striped_matrixの例

コード例 #1

ファイル: mv1.c プロジェクト: jamestoy/parallel

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;
}

コード例 #2

ファイル: matrix_vector_multiplication.c プロジェクト: DmitriyRybalkin/openmpi_samples

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;
}