我首先在cublas中处理矩阵,我已经将它发送到设备并且我想要处理 矩阵的一些列向量,仍然使用cublas函数。我首先尝试使用指针算术运算来偏移设备指针与主机,但它似乎不起作用。 有没有什么办法可以在矩阵中处理矢量而不将其复制回主机。
例如: cublasSscal(int n,float alpha,float * x,int incx); 用于缩放矢量 我有一个设备指针指向列主矩阵B, 我希望缩放B的第三列并且不将向量复制回主机,该怎么做?
答案 0 :(得分:3)
m是行数, ldB是B的主要维度。
cublasSscal (m, alpha, B + 2*ldB, 1); // indices are 0 based