forked from mindspore-Ecosystem/mindspore
!15308 [CPU]reupgrade onednn to V2.1
From: @zuochuanyong Reviewed-by: @zhaizhiqiang,@guoqi1024 Signed-off-by: @guoqi1024
This commit is contained in:
commit
a1e2b966a0
|
@ -2,22 +2,22 @@ set(onednn_CXXFLAGS "-D_FORTIFY_SOURCE=2 -O2")
|
|||
set(onednn_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")
|
||||
if(CMAKE_SYSTEM_NAME MATCHES "Windows")
|
||||
mindspore_add_pkg(onednn
|
||||
VER 1.6
|
||||
VER 2.1
|
||||
LIBS dnnl mkldnn
|
||||
HEAD_ONLY ./include
|
||||
RELEASE on
|
||||
URL https://github.com/oneapi-src/oneDNN/releases/download/v1.6/dnnl_win_1.6.0_cpu_vcomp.zip
|
||||
MD5 fe660e34e9f73ab13a65987819a0712e)
|
||||
URL https://github.com/oneapi-src/oneDNN/releases/download/v2.1/dnnl_win_2.1.0_cpu_vcomp.zip
|
||||
MD5 b3111c4851dad06f7a796b27083dffa8)
|
||||
else()
|
||||
if(ENABLE_GITEE)
|
||||
set(REQ_URL "https://gitee.com/mirrors/MKL-DNN/repository/archive/v2.1.2.tar.gz")
|
||||
set(MD5 "d98f171d7e66e252c79e2e167ba4a8e8")
|
||||
set(REQ_URL "https://gitee.com/mirrors/MKL-DNN/repository/archive/v2.1.tar.gz")
|
||||
set(MD5 "f4c10ad4197ce2358ad1a917e84c288c")
|
||||
else()
|
||||
set(REQ_URL "https://github.com/oneapi-src/oneDNN/archive/v2.1.2.tar.gz")
|
||||
set(MD5 "1df4f16f650b7ea08610a10af013faa3")
|
||||
set(REQ_URL "https://github.com/oneapi-src/oneDNN/archive/v2.1.tar.gz")
|
||||
set(MD5 "2ed85f2c0c3771a7618db04a9e08ae57")
|
||||
endif()
|
||||
mindspore_add_pkg(onednn
|
||||
VER 2.1.2
|
||||
VER 2.1
|
||||
LIBS dnnl mkldnn
|
||||
URL ${REQ_URL}
|
||||
MD5 ${MD5}
|
||||
|
|
Loading…
Reference in New Issue