代码之家 › 专栏 › 技术社区 › Luca Iliesiu

使用cuBlas高效地将矩阵与转置相乘

cublas matrix-multiplication linear-algebra

0

Luca Iliesiu · 技术社区 · 10 年前

当将一个大的密集矩阵与其转置相乘时,有没有一种有效的方法来使用cuBlas?具体来说,是否有任何函数可以利用结果矩阵是对称的这一事实,从而将乘法次数减少一个因子~2。

1 回复 | 直到 10 年前

1

Robert Crovella 10 年前

标准BLAS API ?syrk 是你需要的。

C = alpha * A * A^T + beta * C

对于cuBlas,API名称为 cublas<t>syrk 有关更多详细信息,请参阅cuBlas文档