Version.h 638 B

1234567891011121314151617181920212223
  1. #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
  2. #include <ATen/Context.h>
  3. namespace at {
  4. /// Returns a detailed string describing the configuration PyTorch.
  5. TORCH_API std::string show_config();
  6. TORCH_API std::string get_mkl_version();
  7. TORCH_API std::string get_mkldnn_version();
  8. TORCH_API std::string get_openmp_version();
  9. TORCH_API std::string get_cxx_flags();
  10. TORCH_API std::string get_cpu_capability();
  11. } // namespace at
  12. #else
  13. #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
  14. #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)