| 53 | // calculate single matrix eig in cusolver |
| 54 | template <typename T> |
| 55 | inline void single_eig_cusolver( |
| 56 | DeviceSolverDnHandle& cusolver_H, SingleEigParam& param, |
| 57 | DeviceDnTen<T>& mat, DeviceDnTen<T>& W, |
| 58 | void* buffer, void* buffer_host, int* info, |
| 59 | const size_s mat_size, const size_t buffer_size, const size_t buffer_size_host, |
| 60 | const int mat_offset = 0, const int W_offset = 0, |
| 61 | const int info_offset = 0 |
| 62 | ) { |
| 63 | CHECK_CUDA( cudaSetDevice(cusolver_H.gpu_id) ); |
| 64 | CHECK_CUSOLVER( cusolverDnXsyevd( |
| 65 | cusolver_H.cusolver_dn_handle, param.param, param.jobz, param.uplo, |
| 66 | mat_size, CudaTypeMapper<T>::value, mat.vals + mat_offset, |
| 67 | mat_size, CudaTypeMapper<T>::value, W.vals + W_offset, |
| 68 | CudaTypeMapper<T>::value, |
| 69 | buffer, buffer_size, |
| 70 | buffer_host, buffer_size_host, |
| 71 | info + info_offset |
| 72 | ) ); |
| 73 | } |
| 74 | #endif // DENSE_EIG_H |