使用blas做矩阵乘法

最新推荐文章于 2024-09-11 09:30:23 发布

转载

最新推荐文章于 2024-09-11 09:30:23 发布 · 686 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/huashiyiqike/p/3871927.html

文章标签：

#人工智能 #c/c++ #runtime

本文介绍了如何使用g++编译器结合OpenBLAS库实现矩阵乘法。通过调整参数可以编译不同版本的OpenBLAS。矩阵乘法的转置性质也被提及，指出交换矩阵顺序可以直接得到转置乘法结果。此外，还提到了CUBLAS库中处理向量加法的cblas_daxpy函数及其初始化和销毁过程。

#define min(x,y) (((x) < (y)) ? (x) : (y))

#include <stdio.h>
#include <stdlib.h>
#include <cublas_v2.h>
#include <iostream>
#include <vector>
//extern "C"
//{
    
    
   #include <cblas.h>
//}

using namespace std;
int main()
{

    const enum CBLAS_ORDER Order=CblasRowMajor;
    const enum CBLAS_TRANSPOSE TransA=CblasNoTrans;
    const enum CBLAS_TRANSPOSE TransB=CblasNoTrans;
    const int M=4;//A的行数，C的行数
    const int N=2;//B的列数，C的列数
    const int K=3;//A的列数，B的行数
    const float alpha=1;
    const float beta=0;
    const int lda=K;//A的列
    const int ldb=N;//B的列
    const int ldc=N;//C的列
    const float A[M*K]={
   
   1,2,3,4,5,6,7,8,9,8,7,6};
    const float B[K*N]={
   
   5,4,