1 #pragma once 2 3 #include <c10/util/Exception.h> 4 5 #include <ostream> 6 #include <string> 7 8 namespace at { 9 10 enum class BlasBackend : int8_t { Cublas, Cublaslt }; 11 BlasBackendToString(at::BlasBackend backend)12inline std::string BlasBackendToString(at::BlasBackend backend) { 13 switch (backend) { 14 case BlasBackend::Cublas: 15 return "at::BlasBackend::Cublas"; 16 case BlasBackend::Cublaslt: 17 return "at::BlasBackend::Cublaslt"; 18 default: 19 TORCH_CHECK(false, "Unknown blas backend"); 20 } 21 } 22 23 inline std::ostream& operator<<(std::ostream& stream, at::BlasBackend backend) { 24 return stream << BlasBackendToString(backend); 25 } 26 27 } // namespace at 28