BLAS:gemm vs. gemv

Question

为什么BLAS具有gemm矩阵 - 矩阵乘法gemv功能和矩阵向量乘法的单独函数？矩阵向量乘法不是矩阵乘法的一个特例,其中一个矩阵只有一个行/列吗？

Answer 1

在数学上,矩阵向量乘法是矩阵 - 矩阵乘法的一种特殊情况,但在软件库中实现的并不一定如此.

他们支持不同的选择.例如,gemv支持跨越对其运行的向量的访问,gemm而不支持跨步矩阵布局.在C语言绑定中,gemm要求您指定所有三个矩阵的存储顺序,而gemv对于向量参数则不需要,因为它没有意义.

除了支持不同的选项外,还有一系列可能执行的优化gemm不适用于gemv.如果您知道自己正在使用矩阵向量产品,那么在切换到针对该情况优化的代码路径之前,您不希望库浪费时间来确定情况; 你宁可直接调用它.

Answer 2

当您优化 gemv 和 gemm 时，应用不同的技术：

• 对于矩阵-矩阵运算，您使用的是阻塞算法。块大小取决于缓存大小。
• 为了优化矩阵向量积，您可以使用所谓的融合 1 级运算（例如融合点积或融合 axpy）。

如果您想了解更多详细信息，请告诉我。