Skip to content

Commit

Permalink
Add define for double getrf, getri
Browse files Browse the repository at this point in the history
  • Loading branch information
Liang Zhao committed Nov 7, 2016
1 parent e802471 commit 16307aa
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions paddle/cuda/src/hl_cuda_cublas.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 16307aa

Please sign in to comment.