主机代码中的cuda偏移设备指针

时间:2010-08-17 21:39:48

标签: pointers cuda device

我首先在cublas中处理矩阵,我已经将它发送到设备并且我想要处理 矩阵的一些列向量,仍然使用cublas函数。我首先尝试使用指针算术运算来偏移设备指针与主机,但它似乎不起作用。 有没有什么办法可以在矩阵中处理​​矢量而不将其复制回主机。

例如:     cublasSscal(int n,float alpha,float * x,int incx); 用于缩放矢量     我有一个设备指针指向列主矩阵B,     我希望缩放B的第三列并且不将向量复制回主机,该怎么做?

1 个答案:

答案 0 :(得分:3)

m是行数, ldB是B的主要维度。

cublasSscal (m, alpha, B + 2*ldB, 1); //  indices are 0 based