CUDA总结：CUBLAS_cublassgemm_v2 支持那些类型

CUBLAS_Library.pdf

函数命名规则：类型

<type> <t> Meaning
float, ‘s’ or ‘S’, real single-precision
double, ‘d’ or ‘D’, real double-precision
cuComplex, ‘c’ or ‘C’, complex single-precision
cuDoubleComplex, ‘z’ or ‘Z’, complex double-precision
1
2
3
4
5

这四个类型还可以混合使用，代表源、目的操作数，如Sc、Cs、Dz、Zd等

如矩阵乘法

#define cublasSgemv          cublasSgemv_v2   //float类型
#define cublasDgemv          cublasDgemv_v2   //double类型
#define cublasCgemv          cublasCgemv_v2   //复数float类型
#define cublasZgemv          cublasZgemv_v2   //复数double类型
1
2
3
4