Searched refs:Cusolver (Results 1 – 7 of 7) sorted by relevance
10 enum class LinalgBackend : int8_t { Default, Cusolver, Magma }; enumerator16 case LinalgBackend::Cusolver: in LinalgBackendToString()
253 TORCH_CHECK((b != at::LinalgBackend::Cusolver) || hasCuSOLVER(), in setLinalgPreferredBackend()
385 ? at::LinalgBackend::Cusolver
148 "cusolver": torch._C._LinalgBackend.Cusolver,
1110 case at::LinalgBackend::Cusolver: in ldl_factor_kernel()1244 case at::LinalgBackend::Cusolver: in _cholesky_solve_helper_cuda()1358 case at::LinalgBackend::Cusolver: in cholesky_kernel()1439 case at::LinalgBackend::Cusolver: in cholesky_inverse_kernel_impl()1634 if (preferred_backend == at::LinalgBackend::Cusolver) { in lu_factor()1872 case at::LinalgBackend::Cusolver: in geqrf_kernel()2001 case at::LinalgBackend::Cusolver: in linalg_eigh_kernel()2492 if (preferred_backend == at::LinalgBackend::Cusolver) { in lu_solve_kernel()2719 case at::LinalgBackend::Cusolver: in gels_looped()
2055 .value("Cusolver", at::LinalgBackend::Cusolver) in initModule()
1252 Cusolver: _LinalgBackend