cuda - 如何更改矩阵维度 MatrixMul CUDA 示例

Question

嗨，我尝试通过在启动时传递参数来更改 matrixmul 中的矩阵尺寸。不幸的是，它只接受一些值，如 160*160、320*320、320*1280 640*640，但不接受较低的值，如 80*80 或其他值，如 56*56。你能告诉我为什么以及该怎么做吗？谢谢！

Answer 1

该示例仅演示了如何在 CUDA 中完成 mat 乘法。使用该代码来做真正的 mat mul 不是一个好的选择。相反，您可以使用 cuBlas 库提供的 BLAS 函数，它支持任意维度。

另一方面，如果您想了解有关 mat mul 代码如何工作的更多信息，您可以在 Q 中附加一些代码。