Repository URL to install this package:
|
Version:
2.1.2+cpu ▾
|
#pragma once
#include <string>
#include <c10/macros/Export.h>
namespace at::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::cuda