Я пытаюсь использовать cublasSaxpy
функция из кубла.
Посмотрите на код:
#include <cublas_v2.h>
float *dev_B;
float *dev_tmp;
cublasStatus_t stat ;
cublasHandle_t handle ;
...
...
stat = cublasCreate(&handle);
if(stat != CUBLAS_STATUS_SUCCESS)
{
cout << "CUBLAS Initialization failed!" << endl;
exit(EXIT_FAILURE);
}
...
...
stat = cublasSetVector(N, sizeof(float), vB, N, dev_B, N);
if(stat != CUBLAS_STATUS_SUCCESS)
{
cout << "CUBLAS setting vector failed" << endl;
exit(EXIT_FAILURE);
}
stat = cublasSetVector(N, sizeof(float), nullvec, N, dev_tmp, N);
if(stat != CUBLAS_STATUS_SUCCESS)
{
cout << "CUBLAS setting vector failed" << endl;
exit(EXIT_FAILURE);
}
...
...
stat = cublasSaxpy(handle, N, 1.0f, dev_B, 1, dev_tmp, 1);
Итак, я использую eclipse nsight из CUDA 5.0 RC1. Я не создавал проект, потому что я не закончил его, но среда IDE подчеркивает чтение в операции saxpy:
Invalid arguments '
Candidates are:
enum {cublas_api.h:3039} cublasSaxpy_v2(cublasContext *, int, const float *, const float *, int, float *, int)
В API CUBLAS V2 все скалярные вещественные аргументы передаются по ссылке, а не по значению. Правильный вызов для saxpy
будет что-то вроде:
const float alpha = 1.0f;
stat = cublasSaxpy(handle, N, &alpha, dev_B, 1, dev_tmp, 1);
Это четко обсуждается в документации.
Других решений пока нет …