#include<omp.h>
#include<stdio.h>
#include<time.h>
#include <stdlib.h>
#define NN 1000
long long int a[NN][NN], b[NN][NN];
long long c[NN][NN];
void solve(int n,int num_thread)
{
int i, j,k;
omp_set_num_threads(num_thread);
#pragma omp parallel shared(a,b,c) private (i,j,k)
{
#pragma omp for schedule (dynamic)
for (i=0;i<n;i++)
{
for (j=0;j<n;j++)
{
c[i][j]=0;
for (k=0;k<n;k++)
{
c[i][j]+=a[i][k]*b[k][j];
}
}
}
}
}
int main(int arg1, char *arg2[])
{
int n,num_thread,N,T;
int i, j;
FILE *fp;
clock_t startTime, endTime;
double time;
num_thread=atoi(arg2[2]);
N=atoi(arg2[1]);
n=1000;
fp=fopen("matrix.txt","r");
for(i=0;i<n;i++)
for(j=0;j<n;j++)
fscanf(fp,"%I64d",&a[i][j]);
fclose(fp);
for (i=0;i<n;i++)
for (j=0;j<n;j++)
b[i][j]=a[i][j];
startTime=clock();
for (T=1;T<=N;T++)
{
solve(n,num_thread);
for (i=0;i<n;i++)
for (j=0;j<n;j++)
{
b[i][j]=c[i][j];
}
}
/* for (i=0;i<NN;i++)
{
for (j=0;j<NN;j++)
printf("%I64d ",b[i][j]);
printf("\n");
}
*/
endTime=clock();
time=endTime-startTime;
printf("time=%lfms\n",time);
return 0;
}
评论0