CUDA总结:CUBLAS

CUBLAS_Library.pdf

函数命名规则:类型

  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

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

如矩阵乘法

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

你可能感兴趣的:(cuda)