1

cublasSaxpycublasの関数を使用しようとしています。

コードを見てください:

#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)
4

1 に答える 1

3

CUBLAS V2 API では、すべてのスカラー実数引数は、値ではなく参照によって渡されます。の正しい呼び出しは次のsaxpyようになります。

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

これはドキュメントで明確に説明されています。

于 2012-09-15T05:28:20.563 に答える