我想使用 Openmp 在四个线程之间分配任务。我有 20x20 矩阵,任务应该平均分布在四个线程中,以下程序给出错误结果。我认为它需要在 for 循环中更改参数,任何人都可以帮助我。
#include<stdio.h>
#include<omp.h>
#include "head.h"
//int sum=0;
int c[20][20];
//#include<conio.h>
int main(void) {
int A[20][20],B[20][20],C[20][20];
int i;
static int j,e;
static sum=0;
FILE *fp;
unsigned long long a,b,c,d;
int threadno;
fp=fopen("m2.txt","w");
// printf("\n%d \n",h[20][20]);
#pragma omp parallel shared(a,b,c,d)
{
threadno=omp_get_thread_num();
if(threadno==0)
{
for (i=0;i<5;i++)
{
for (j=0;j<5;j++)
{
A[i][j]=i*j;
B[i][j]=i*j;
}
}
for (i=0;i<5;i++)
{
for (j=0;j<5;j++)
{
for (e=0;e<5;e++)
{
sum+=A[i][e]*B[e][j];
C[i][j]=sum;
}
}
}
}
else if(threadno==1)
{
for (i=5;i<10;i++)
{
for (j=5;j<10;j++)
{
A[i][j]=i*j;
B[i][j]=i*j;
}
}
for (i=5;i<10;i++)
{
for (j=5;j<10;j++)
{
for (e=5;e<10;e++)
{
sum+=A[i][e]*B[e][j];
C[i][j]=sum;
}
}
}
}
else if(threadno==2)
{
for (i=10;i<15;i++)
{
for (j=10;j<15;j++)
{
A[i][j]=i*j;
B[i][j]=i*j;
}
}
for (i=10;i<15;i++)
{
for (j=10;j<15;j++)
{
for (e=10;e<15;e++)
{
sum+=A[i][e]*B[e][j];
C[i][j]=sum;
}
}
}
}
else if(threadno==3)
{
for (i=15;i<20;i++)
{
for (j=15;j<20;j++)
{
A[i][j]=i*j;
B[i][j]=i*j;
}
}
for (i=15;i<20;i++)
{
for (j=15;j<20;j++)
{
for (e=15;e<20;e++)
{
sum+=A[i][e]*B[e][j];
C[i][j]=sum;
}
}
}
}
for (i=0;i<20;i++){
for (j=0;j<20;j++) {
fprintf(fp,"%d \t",C[i][j]);
}
}
}
fclose(fp);
}