使用库布拉斯萨克斯时出错

Error while using cublas saxpy

本文关键字：出错 萨克斯 布拉斯      更新时间：2023-10-16

我试图使用cublas的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);

所以，我正在使用CUDA 5.0 RC1中的eclipse nsight。我没有构建项目，因为我没有完成它，但是 IDE 在 saxpy 操作中给了我一个读取下划线：

Invalid arguments '
Candidates are:
enum {cublas_api.h:3039} cublasSaxpy_v2(cublasContext *, int, const float *, const float *, int, float *, int)

在 CUBLAS V2 API 中，所有标量实数参数都是通过引用传递的，而不是通过值传递的。对saxpy的正确调用是这样的：

const float alpha = 1.0f;
stat = cublasSaxpy(handle, N, &alpha, dev_B, 1, dev_tmp, 1);

这在文档中有明确的讨论。

相关文章：