OpenMP实现矩阵相乘加速

参考
https://blog.csdn.net/lusongno1/article/details/82929596
工具文档
https://www.hz-bin.cn/OpenMP

在OpenMP中
omp_get_num_procs()函数来获取默认最大可用线程个数,
omp_get_thread_num()函数来获得每个线程的ID,为了使用这两个函数,我们需要include <omp.h>
另外,omp_get_num_threads()返回当前并行区域中的活动线程个数,如果在并行区域外部调用,返回1。
omp_set_num_threads(5)设置进入并行区域时,将要创建的线程个数为5个。在指导语句中直接设置核心数也是可以的:#pragma omp parallel num_threads(6)

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <omp.h>
#include<iostream>
#define m 2000

int a[m][m];
int b[m][m];
int c[m][m];

int main() {
    clock_t start, end;


    int i, j, k;
    //  srand((unsigned)time(NULL));//根据时间来产生随机数种子
    srand(0);
    for (i = 0; i < m; i++) {
        for (j = 0; j < m; j++) {
            a[i][j] = (int)rand() / (RAND_MAX);//产生[0,1]随机数
            b[i][j] = (int)rand() / (RAND_MAX);
            c[i][j] = 0;//初始化c矩阵用来存结果
#if m <= 1   //打印一下呗 
            printf("%f ", a[i][j]);
            if (j == m - 1) {
                printf("\n");
            }
#endif
        }
    }
    //  int gettimeofday(struct timeval*tv, struct timezone *tz);


    omp_set_num_threads(8);
    int pnum = omp_get_num_procs();//获取系统默认最大线程数量
    fprintf(stderr, "Thread_pnum = %d\n", pnum);
    //  gettimeofday(&st, NULL);
    start = clock();

#pragma omp parallel shared(a,b,c) private(j,k)
    {
#pragma omp for  nowait //schedule(static)
        for (i = 0; i < m; i++) {

            for (j = 0; j < m; j++) {
                
                for (k = 0; k < m; k++) {
                    c[i][j] = c[i][j] + a[i][k] * b[k][j];
                }
                
            }

        }
        std::cout <<"thread id: "<< omp_get_thread_num() <<" time: "<< clock() << "\n";
    }
    end = clock();
    printf("matrix multiply time: %0.6lf sec\n", ((double)end - start) / CLK_TCK);

    return 0;
}

使用nowait语句


image.png

没有使用nowait语句


image.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 目前很多移动端的深度学习前向运算框架都用到openmp,如ncnn、paddlelite、mace等。所以这篇来介...
    半笔闪阅读 2,494评论 0 2
  • OpenMP OpenMP是一种支持多平台的共享内存、多处理器(多线程)的规范与API。OpenMP的API包括编...
    zackary_shen阅读 4,457评论 0 0
  • 博客地址:https://www.hz-bin.cn/OpenMP[https://www.hz-bin.cn/O...
    binzjut阅读 40,675评论 3 20
  • 1.定义 使用parallel指令只是产生了并行域,让多个线程分别执行相同的任务,并没有实际的使用价值。paral...
    星夜兼程工作笔记阅读 3,427评论 0 0
  • OpenMP2.5 有底层API后,就已经可以实现并行编程;然而,很多时候串行算法已经成型,如果继续使用原有的底层...
    一梦换须臾_阅读 1,439评论 0 0