| namespace at { | |
| enum class BlasBackend : int8_t { Cublas, Cublaslt }; | |
| inline std::string BlasBackendToString(at::BlasBackend backend) { | |
| switch (backend) { | |
| case BlasBackend::Cublas: | |
| return "at::BlasBackend::Cublas"; | |
| case BlasBackend::Cublaslt: | |
| return "at::BlasBackend::Cublaslt"; | |
| default: | |
| TORCH_CHECK(false, "Unknown blas backend"); | |
| } | |
| } | |
| inline std::ostream& operator<<(std::ostream& stream, at::BlasBackend backend) { | |
| return stream << BlasBackendToString(backend); | |
| } | |
| } // namespace at | |