JSX_TTS / torch /include /ATen /cuda /llvm_jit_strings.h
UMMJ's picture
Upload 5875 files
9dd3461
raw
history blame contribute delete
434 Bytes
#pragma once
#include <string>
#include <c10/macros/Export.h>
namespace at {
namespace cuda {
TORCH_CUDA_CPP_API const std::string &get_traits_string();
TORCH_CUDA_CPP_API const std::string &get_cmath_string();
TORCH_CUDA_CPP_API const std::string &get_complex_body_string();
TORCH_CUDA_CPP_API const std::string &get_complex_half_body_string();
TORCH_CUDA_CPP_API const std::string &get_complex_math_string();
}} // namespace at