diff --git a/aten/src/ATen/ParallelCommon.cpp b/aten/src/ATen/ParallelCommon.cpp index 49b83d9157d..3e86fb47282 100644 --- a/aten/src/ATen/ParallelCommon.cpp +++ b/aten/src/ATen/ParallelCommon.cpp @@ -62,7 +62,9 @@ std::string get_parallel_info() { ss << "\tomp_get_max_threads() : " << omp_get_max_threads() << '\n'; #endif +#if defined(__x86_64__) || defined(_M_X64) ss << at::get_mkl_version() << '\n'; +#endif #if AT_MKL_ENABLED() ss << "\tmkl_get_max_threads() : " << mkl_get_max_threads() << '\n'; #endif @@ -75,8 +77,10 @@ std::string get_parallel_info() { ss << "Environment variables:" << '\n'; ss << "\tOMP_NUM_THREADS : " << get_env_var("OMP_NUM_THREADS", "[not set]") << '\n'; +#if defined(__x86_64__) || defined(_M_X64) ss << "\tMKL_NUM_THREADS : " << get_env_var("MKL_NUM_THREADS", "[not set]") << '\n'; +#endif ss << "ATen parallel backend: "; #if AT_PARALLEL_OPENMP