Version.h 384 B

123456789101112131415161718
  1. #include <ATen/Context.h>
  2. namespace at {
  3. /// Returns a detailed string describing the configuration PyTorch.
  4. TORCH_API std::string show_config();
  5. TORCH_API std::string get_mkl_version();
  6. TORCH_API std::string get_mkldnn_version();
  7. TORCH_API std::string get_openmp_version();
  8. TORCH_API std::string get_cxx_flags();
  9. TORCH_API std::string get_cpu_capability();
  10. } // namespace at