如何更改矩阵尺寸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中附上一些代码。