QMCPACK
cuBLAS_LU.hpp File Reference

At the qmcplusplus cuBLAS_LU level all *, **, *[] are assumed to be to device addresses. More...

+ Include dependency graph for cuBLAS_LU.hpp:
+ This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 qmcplusplus
 helper functions for EinsplineSetBuilder
 
 qmcplusplus::cuBLAS_LU
 

Functions

template<typename T >
void computeInverseAndDetLog_batched (cublasHandle_t &h_cublas, cudaStream_t &hstream, const int n, const int lda, T *Ms[], T *Cs[], T *LU_diags, int *pivots, int *host_infos, int *infos, std::complex< double > *log_dets, const int batch_size)
 Takes PsiM in column major layout and uses LU factorization to compute the log determinant and invPsiM. More...
 
template<typename T >
void computeGetrf_batched (cublasHandle_t &h_cublas, cudaStream_t &hstream, const int n, const int lda, T *Ms[], int *pivots, int *host_infos, int *infos, const int batch_size)
 
template<typename T >
void computeLogDet_batched (cudaStream_t &hstream, const int n, const int lda, T **Ms, const int *pivots, std::complex< double > *logdets, const int batch_size)
 
template<typename T >
void computeGetri_batched (cublasHandle_t &h_cublas, cudaStream_t &hstream, const int n, const int lda, T *Ms[], T *Cs[], int *pivots, int *host_infos, int *infos, const int batch_size)
 
template void computeInverseAndDetLog_batched< double > (cublasHandle_t &h_cublas, cudaStream_t &hstream, const int n, const int lda, double *Ms[], double *Cs[], double *LU_diags, int *pivots, int *host_infos, int *infos, std::complex< double > *log_dets, const int batch_size)
 
template void computeInverseAndDetLog_batched< std::complex< double > > (cublasHandle_t &h_cublas, cudaStream_t &hstream, const int n, const int lda, std::complex< double > *Ms[], std::complex< double > *Cs[], std::complex< double > *LU_diags, int *pivots, int *host_infos, int *infos, std::complex< double > *log_dets, const int batch_size)
 

Detailed Description

At the qmcplusplus cuBLAS_LU level all *, **, *[] are assumed to be to device addresses.

Definition in file cuBLAS_LU.hpp.