mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

onednn on windows

This commit is contained in:
xulei 2022-02-08 10:25:45 +08:00
parent 990614d133
commit 80a5930e6e
2 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmake/external_libs/mkl_dnn.cmake
View File

@ -11,8 +11,8 @@ if(CMAKE_SYSTEM_NAME MATCHES "Windows")
LIBS dnnl mkldnn
HEAD_ONLY ./include
RELEASE on
URL https://github.com/oneapi-src/oneDNN/releases/download/v2.2/dnnl_win_2.2.0_cpu_vcomp.zip
MD5 fa12c693b2ec07700d174e1e99d60a7e)
URL http://tools.mindspore.cn/libs/dnnl/dnnl_win_2.2.0_cpu_vcomp.zip
MD5 139fcdbd601a970fb86dd15b30ba5ae3)
else()
if(ENABLE_GITEE)
set(REQ_URL "https://gitee.com/mirrors/MKL-DNN/repository/archive/v2.2.tar.gz")

5
mindspore/ccsrc/backend/graph_compiler/transform.cc
View File

@ -615,11 +615,6 @@ void SetMindRTEnable() {
return;
}
#if defined(_WIN32) || defined(_WIN64)
context_ptr->set_param<bool>(MS_CTX_ENABLE_MINDRT, false);
return;
#endif
MS_LOG(DEBUG) << "Enable mindRT.";
context_ptr->set_param<bool>(MS_CTX_ENABLE_MINDRT, true);
}