namespace at { | |
enum class LinalgBackend : int8_t { Default, Cusolver, Magma }; | |
inline std::string LinalgBackendToString(at::LinalgBackend backend) { | |
switch (backend) { | |
case LinalgBackend::Default: | |
return "at::LinalgBackend::Default"; | |
case LinalgBackend::Cusolver: | |
return "at::LinalgBackend::Cusolver"; | |
case LinalgBackend::Magma: | |
return "at::LinalgBackend::Magma"; | |
default: | |
TORCH_CHECK(false, "Unknown linalg backend"); | |
} | |
} | |
inline std::ostream& operator<<( | |
std::ostream& stream, | |
at::LinalgBackend backend) { | |
return stream << LinalgBackendToString(backend); | |
} | |
} // namespace at | |