Как использовать библиотеку CUBLAS в функции шаблона?

Question

Как использовать библиотеку CUBLAS в функции шаблона?

CUBLAS имеет отдельная функция для каждого типа данных, но я хочу вызвать CUBLAS из шаблона, например:

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

Как мне вызвать правильный вызов функции?

1

c++cublas templates

Решение

Другие решения

Других решений пока нет …

Источник

Accepted Answer

Я написал функции-обертки для разных типов с одинаковыми именами.

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, ...);
...
}

Это перегрузка во время завершения и вообще не требует затрат времени выполнения, хотя вам нужно написать длинный список для функций-оболочек.

2