OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[BE] Only print MKL version on x86 platforms (#143763)

Browse Source 
As it will obviously be missing on ARM/S390, etc

Test plan: run `python3 -c "import torch;print(torch.__config__.parallel_info())"` on both x86 and non-x86 system
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143763
Approved by: https://github.com/Skylion007, https://github.com/albanD
This commit is contained in:
Nikita Shulga 2024-12-24 02:04:23 +00:00 committed by PyTorch MergeBot
parent 7d1c666139
commit da21fabf34
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
aten/src/ATen/ParallelCommon.cpp
View File

@ -62,7 +62,9 @@ std::string get_parallel_info() {
ss << "\tomp_get_max_threads() : " << omp_get_max_threads() << '\n'; ss << "\tomp_get_max_threads() : " << omp_get_max_threads() << '\n';
#endif #endif
#if defined(__x86_64__) || defined(_M_X64)
ss << at::get_mkl_version() << '\n'; ss << at::get_mkl_version() << '\n';
#endif
#if AT_MKL_ENABLED() #if AT_MKL_ENABLED()
ss << "\tmkl_get_max_threads() : " << mkl_get_max_threads() << '\n'; ss << "\tmkl_get_max_threads() : " << mkl_get_max_threads() << '\n';
#endif #endif
@ -75,8 +77,10 @@ std::string get_parallel_info() {
ss << "Environment variables:" << '\n'; ss << "Environment variables:" << '\n';
ss << "\tOMP_NUM_THREADS : " ss << "\tOMP_NUM_THREADS : "
<< get_env_var("OMP_NUM_THREADS", "[not set]") << '\n'; << get_env_var("OMP_NUM_THREADS", "[not set]") << '\n';
#if defined(__x86_64__) || defined(_M_X64)
ss << "\tMKL_NUM_THREADS : " ss << "\tMKL_NUM_THREADS : "
<< get_env_var("MKL_NUM_THREADS", "[not set]") << '\n'; << get_env_var("MKL_NUM_THREADS", "[not set]") << '\n';
#endif
ss << "ATen parallel backend: "; ss << "ATen parallel backend: ";
#if AT_PARALLEL_OPENMP #if AT_PARALLEL_OPENMP