Skip to content

Commit 4905751

Browse files
lzhao4everemailweixu
authored andcommitted
Add define for double getrf, getri (#381)
1 parent e802471 commit 4905751

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/cuda/src/hl_cuda_cublas.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,8 @@ DYNAMIC_LOAD_CUBLAS_WRAP(cublasCgemmBatched)
7878
DYNAMIC_LOAD_CUBLAS_WRAP(cublasZgemmBatched)
7979
DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgetrfBatched)
8080
DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgetriBatched)
81+
DYNAMIC_LOAD_CUBLAS_WRAP(cublasDgetrfBatched)
82+
DYNAMIC_LOAD_CUBLAS_WRAP(cublasDgetriBatched)
8183
CUBLAS_BLAS_ROUTINE_EACH(DYNAMIC_LOAD_CUBLAS_V2_WRAP)
8284

8385
#undef DYNAMIC_LOAD_CUBLAS_WRAP

0 commit comments

Comments
 (0)