xref: /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/llvm_jit_strings.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
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