logo资料库

矩阵乘法 mpi.docx

第1页 / 共4页
第2页 / 共4页
第3页 / 共4页
第4页 / 共4页
资料共4页,全文预览结束
#include #include #include #include "mpi.h" #define N 3 int main(int argc,char **argv) { MPI_Init(&argc,&argv); int rank,size; MPI_Comm_rank(MPI_COMM_WORLD,&rank); MPI_Comm_size(MPI_COMM_WORLD,&size); int m=atoi(argv[1]); int n=atoi(argv[2]); int block=(m+size-1)/size*n; if(rank==size-1 && block>m*n-(size-1)*block) block=m*n-(size-1)*block; int *A=NULL; int *TA=(int*)malloc(block*sizeof(int)); int **B=(int**)malloc(m*sizeof(int*)); int *counts=NULL; int *ofss=NULL; int i,j; for(i=0;i
for(i=0;i
int k; for(k=0;k
{ printf("[\t"); for(j=0;j
分享到:
收藏