如何在模板函數中使用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時間過載和運行時沒有性價比可言，但你必須寫的包裝功能的長列表。