CULA矩阵相乘和CUBLAS矩阵相乘

CULA的矩阵相乘:

culaDeviceDgemm('N','N',n,m,k,alpha,

b_device,n,           //b   k  x  n
a_device,k,           //a   m  x  k
beta,

c_device,n);

上式表示:C=A*B的矩阵相乘方法,而且数据类型为double,也可以使用float类型数据的函数:culaDeviceSgemm,对应float类型数据,'N'表示不进行转置,'T'表示A或B转置相乘

其中:

b_device是B矩阵,大小为k x n;

a_device是A矩阵,大小为m x k;

c_device是C矩阵,大小为m x n;

double alpha=1;
double beta=0;


CUBLAS矩阵相乘:

cublasDgemm(handle,CUBLAS_OP_N,CUBLAS_OP_N,n,m,k,&alpha,

           b_device,n,

           a_device,k,

          &beta,c_device,n);

意义和CULA相同,CUBLAS_OP_N在CULA中对应的是'N',CUBLAS_OP_T在CULA中对应的是'T',也同样对应的有cublasSgemm等函数提供不同的数据类型进行使用

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值