llvm_jit_strings.h 434 B

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