|
|
|
@ -67,20 +67,20 @@ extern void *cublas_dso_handle;
|
|
|
|
|
__macro(cublasSgemm); \
|
|
|
|
|
__macro(cublasDgemm); \
|
|
|
|
|
__macro(cublasSgeam); \
|
|
|
|
|
__macro(cublasDgeam);
|
|
|
|
|
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasCreate);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasDestroy);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasSetStream);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasSetPointerMode);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasGetPointerMode);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgemmBatched);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasDgemmBatched);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasCgemmBatched);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasZgemmBatched);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgetrfBatched);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgetriBatched);
|
|
|
|
|
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasDgetrfBatched);
|
|
|
|
|
__macro(cublasDgeam); \
|
|
|
|
|
__macro(cublasCreate_v2); \
|
|
|
|
|
__macro(cublasDestroy_v2); \
|
|
|
|
|
__macro(cublasSetStream_v2); \
|
|
|
|
|
__macro(cublasSetPointerMode_v2); \
|
|
|
|
|
__macro(cublasGetPointerMode_v2); \
|
|
|
|
|
__macro(cublasSgemmBatched); \
|
|
|
|
|
__macro(cublasDgemmBatched); \
|
|
|
|
|
__macro(cublasCgemmBatched); \
|
|
|
|
|
__macro(cublasZgemmBatched); \
|
|
|
|
|
__macro(cublasSgetrfBatched); \
|
|
|
|
|
__macro(cublasSgetriBatched); \
|
|
|
|
|
__macro(cublasDgetrfBatched); \
|
|
|
|
|
__macro(cublasDgetriBatched)
|
|
|
|
|
|
|
|
|
|
CUBLAS_BLAS_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP);
|
|
|
|
|
|
|
|
|
|