Как использовать библиотеку CUBLAS в функции шаблона?
CUBLAS имеет отдельная функция для каждого типа данных, но я хочу вызвать CUBLAS из шаблона, например:
template <typename T> foo(...) {
...
cublas<S/D/C/Z>geam(..., const T* A, ...);
...
}
Как мне вызвать правильный вызов функции?
Решение
Я написал функции-обертки для разных типов с одинаковыми именами.
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, ...);
...
}
Это перегрузка во время завершения и вообще не требует затрат времени выполнения, хотя вам нужно написать длинный список для функций-оболочек.
Другие решения
Других решений пока нет …