美文网首页
OpenMP实现矩阵相乘加速

OpenMP实现矩阵相乘加速

作者: 吃核桃用手夹 | 来源:发表于2022-03-22 17:10 被阅读0次

参考
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实现矩阵相乘加速

    参考https://blog.csdn.net/lusongno1/article/details/8292959...

  • numpy -- 实现线性代数

    Python 实现线性代数 m_n 与 n_k 这样的矩阵才能相乘 矩阵求逆 矩阵和矩阵的逆相乘结果为单位矩阵 qr分解

  • 矩阵

    1. 线性方程组 2. 矩阵定义 3. 矩阵运算 矩阵的加法矩阵的加法 数与矩阵相乘数与矩阵相乘 矩阵与矩阵相乘矩...

  • OpenMP 多线程加速矩阵乘法

    MatMul测试结果 测试代码 matmul.hpp main.cpp

  • 矩阵运算

    矩阵与标量的乘法 每个元素与标量相乘 矩阵相乘 只有第一个矩阵的列数与第二个矩阵的行数相同,才可以相乘。新矩阵行数...

  • Hadoop Page Rank

    Page Rank实现本质也是矩阵相乘 Idea: 用Transition Matrix 乘 initial po...

  • OpenGL笔记六:纹理常用API(一)

    前言 期待您移步上篇:OpenGL笔记五:综合实例理解-压栈、出栈、堆栈矩阵相乘、矩阵相乘、向量相乘 纹理 纹理只...

  • 矩阵论-矩阵乘法的意义

    推荐书籍: 《神奇矩阵》 AB矩阵相乘的表示形式我喜欢这种。 当然有了矩阵相乘也有其反向操作,矩阵的分解。

  • 矩阵和矩阵的相乘_线性代数_day15

    矩阵在图像变换中的应用 矩阵乘法的定义 矩阵和矩阵的相乘,就是分成每一个矩阵和向量的相乘 矩阵A的列数必须和矩阵B...

  • 矩阵相乘

    /*矩阵相乘 题目描述:两个矩阵的乘法仅当第一个矩阵A的行数和另一个 矩阵B的列数相等时才能定义...

网友评论

      本文标题:OpenMP实现矩阵相乘加速

      本文链接:https://www.haomeiwen.com/subject/oscadrtx.html