1 #pragma once 2 3 #include <string> 4 #include <c10/macros/Export.h> 5 6 namespace at::cuda { 7 8 TORCH_CUDA_CPP_API const std::string &get_traits_string(); 9 TORCH_CUDA_CPP_API const std::string &get_cmath_string(); 10 TORCH_CUDA_CPP_API const std::string &get_complex_body_string(); 11 TORCH_CUDA_CPP_API const std::string &get_complex_half_body_string(); 12 TORCH_CUDA_CPP_API const std::string &get_complex_math_string(); 13 14 } // namespace at::cuda 15