указатель устройства смещения cuda в коде хоста - PullRequest
4 голосов
/ 18 августа 2010

Я сначала обрабатываю матрицу в cublas, я уже отправил ее на устройство и хочу обработать некоторый столбец-вектор матрицы, все еще использующий функцию cublas. Сначала я пытаюсь использовать арифметическую операцию с указателем для смещения указателя устройства от хоста, но, похоже, это не работает Есть ли способ я могу обработать вектор в матрице, не копируя его обратно на хост.

например: cublasSscal (int n, float alpha, float * x, int incx); используется для масштабирования вектора У меня есть указатель устройства, чтобы указать столбец основной матрицы B, я хочу масштабировать третий столбец B и без копирования вектора обратно на хост, как это сделать?

1 Ответ

3 голосов
/ 18 августа 2010

m - количество строк, ldB - начальный размер B.

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