如何在模板函数中使用CUBLAS库？

Question

CUBLAS对每种类型的数据的separate function，但我想一个模板内调用CUBLAS，例如：

template <typename T> foo(...) { 
    ... 
    cublas<S/D/C/Z>geam(..., const T* A, ...); 
    ... 
} 

如何触发了正确的函数调用？

Answer 1

我为具有相同函数名称的不同类型编写了cublas wrapper函数。

inline cublasStatus_t cublasGgeam(cublasHandle_t handle, 
     cublasOperation_t transa, cublasOperation_t transb, 
     int m, int n, 
     const float *alpha, 
     const float *A, int lda, 
     const float *beta, 
     const float *B, int ldb, 
     float *C, int ldc) 
{ 
    return cublasSgeam(handle, transa, transb, m, n, alpha, A, lda, beta, B, ldb, C, ldc); 
} 

inline cublasStatus_t cublasGgeam(cublasHandle_t handle, 
     cublasOperation_t transa, cublasOperation_t transb, 
     int m, int n, 
     const double *alpha, 
     const double *A, int lda, 
     const double *beta, 
     const double *B, int ldb, 
     double *C, int ldc) 
{ 
    return cublasDgeam(handle, transa, transb, m, n, alpha, A, lda, beta, B, ldb, C, ldc); 
} 

之后，您可以调用geam（）作为具有相同函数名称的任何类型。 C++编译器会根据参数的类型选择正确的函数。在你情况下，应该像

template <typename T> foo(...) { 
    ... 
    cublasGgeam(..., A, ...); 
    ... 
} 

这是一个COMPLE时间过载和运行时没有性价比可言，但你必须写的包装功能的长列表。