#include "mpi.h"
#include <stdio.h>
#include <math.h>

#define NUMPROCS 64
#define N 8192
// 行列のサイズ
#define M 128

int main(int argc, char **argv){
  int n, myid, numprocs, i, j;
  double A[N][N], x[N], y[N];
  int f;
  MPI_Status istatus;

  double Stime, Etime;

  MPI_Init(&argc, &argv);
  MPI_Comm_size(MPI_COMM_WORLD, &numprocs);
  MPI_Comm_rank(MPI_COMM_WORLD, &myid);

  for(i=0; i<N; i++){
    for(j=0; j<N; j++){
      A[i][j] = 1;
    }
    x[i] = 1;
  }

  Stime = MPI_Wtime();
  
  for(i=myid*M; i<myid*M+M; i++){
    y[i] = 0;
    for(j=0; j<N; j++){
      y[i] += A[i][j]*x[j];
    }
  }
    // myid*M ~ myid*M+M までのデータを担当する

  Etime = MPI_Wtime();

    // myidが0のプロセッサにデータを集める
  if(myid!=0){
    MPI_Send(y+myid*M, M, MPI_DOUBLE, 0, 0, MPI_COMM_WORLD);
      // 0番でないプロセッサは0番のプロセッサにmyid*M ~ myid*M+M までの結果を送る
  }else{
    for(i=M; i<N; i+=M){
      MPI_Recv(y+i, M, MPI_DOUBLE, i/M, 0, MPI_COMM_WORLD, &istatus);
        // 0番は受信する
    }
    f=1;
    for(i=0; i<N; i++){
      if(fabs(y[i]-N)>1e-7) f=0;
    }

    printf("%lf %lf\n",(Etime-Stime), MPI_Wtick() );
    printf("f=%d\n", f);
  }

  MPI_Barrier(MPI_COMM_WORLD);
  MPI_Finalize();
  
  return 0;
}