Home
last modified time | relevance | path

Searched refs:LinalgBackend (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/aten/src/ATen/
DLinalgBackend.h10 enum class LinalgBackend : int8_t { Default, Cusolver, Magma }; enum
12 inline std::string LinalgBackendToString(at::LinalgBackend backend) { in LinalgBackendToString()
14 case LinalgBackend::Default: in LinalgBackendToString()
16 case LinalgBackend::Cusolver: in LinalgBackendToString()
18 case LinalgBackend::Magma: in LinalgBackendToString()
27 at::LinalgBackend backend) {
DContext.cpp247 at::LinalgBackend Context::linalgPreferredBackend() const { in linalgPreferredBackend()
251 void Context::setLinalgPreferredBackend(at::LinalgBackend b) { in setLinalgPreferredBackend()
253 TORCH_CHECK((b != at::LinalgBackend::Cusolver) || hasCuSOLVER(), in setLinalgPreferredBackend()
255 TORCH_CHECK((b != at::LinalgBackend::Magma) || hasMAGMA(), in setLinalgPreferredBackend()
257 if (b != at::LinalgBackend::Default) { in setLinalgPreferredBackend()
DContext.h217 at::LinalgBackend linalgPreferredBackend() const;
218 void setLinalgPreferredBackend(at::LinalgBackend);
383 at::LinalgBackend linalg_preferred_backend =
385 ? at::LinalgBackend::Cusolver
386 : at::LinalgBackend::Default;
/external/pytorch/aten/src/ATen/native/cuda/linalg/
DBatchLinearAlgebra.cpp1110 case at::LinalgBackend::Cusolver: in ldl_factor_kernel()
1113 case at::LinalgBackend::Magma: in ldl_factor_kernel()
1244 case at::LinalgBackend::Cusolver: in _cholesky_solve_helper_cuda()
1246 case at::LinalgBackend::Magma: in _cholesky_solve_helper_cuda()
1358 case at::LinalgBackend::Cusolver: in cholesky_kernel()
1361 case at::LinalgBackend::Magma: in cholesky_kernel()
1439 case at::LinalgBackend::Cusolver: in cholesky_inverse_kernel_impl()
1441 case at::LinalgBackend::Magma: in cholesky_inverse_kernel_impl()
1634 if (preferred_backend == at::LinalgBackend::Cusolver) { in lu_factor()
1638 if (preferred_backend == at::LinalgBackend::Magma) { in lu_factor()
[all …]
/external/pytorch/torch/csrc/
DModule.cpp2053 py::enum_<at::LinalgBackend>(py_module, "_LinalgBackend") in initModule()
2054 .value("Default", at::LinalgBackend::Default) in initModule()
2055 .value("Cusolver", at::LinalgBackend::Cusolver) in initModule()
2056 .value("Magma", at::LinalgBackend::Magma); in initModule()
2058 py_module.def("_set_linalg_preferred_backend", [](at::LinalgBackend b) { in initModule()
/external/pytorch/aten/src/ATen/native/
DLinearAlgebraUtils.h414 && at::globalContext().linalgPreferredBackend() != at::LinalgBackend::Magma; in svd_uses_cusolver()