Add define for double getrf, getri (#381)

4905751a · lzhao4ever · emailweixu · e802471c · 4905751a
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

paddle/cuda/src/hl_cuda_cublas.cc paddle/cuda/src/hl_cuda_cublas.cc +2 -0

未找到文件。
--- a/paddle/cuda/src/hl_cuda_cublas.cc
+++ b/paddle/cuda/src/hl_cuda_cublas.cc
@@ -78,6 +78,8 @@ DYNAMIC_LOAD_CUBLAS_WRAP(cublasCgemmBatched)
 DYNAMIC_LOAD_CUBLAS_WRAP(cublasZgemmBatched)
 DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgetrfBatched)
 DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgetriBatched)
+DYNAMIC_LOAD_CUBLAS_WRAP(cublasDgetrfBatched)
+DYNAMIC_LOAD_CUBLAS_WRAP(cublasDgetriBatched)
 CUBLAS_BLAS_ROUTINE_EACH(DYNAMIC_LOAD_CUBLAS_V2_WRAP)

 #undef DYNAMIC_LOAD_CUBLAS_WRAP