|
|
|
@@ -2,26 +2,26 @@ 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.5 |
|
|
|
VER 1.6 |
|
|
|
LIBS dnnl mkldnn |
|
|
|
HEAD_ONLY ./include |
|
|
|
RELEASE on |
|
|
|
URL https://github.com/oneapi-src/oneDNN/releases/download/v1.5/dnnl_win_1.5.0_cpu_vcomp.zip |
|
|
|
MD5 17757c84f49edd42d34ae8c9288110a1) |
|
|
|
URL https://github.com/oneapi-src/oneDNN/releases/download/v1.6/dnnl_win_1.6.0_cpu_vcomp.zip |
|
|
|
MD5 fe660e34e9f73ab13a65987819a0712e) |
|
|
|
else() |
|
|
|
if (ENABLE_GITEE) |
|
|
|
set(REQ_URL "https://gitee.com/mirrors/MKL-DNN/repository/archive/v1.5.tar.gz") |
|
|
|
set(MD5 "5e0f3800d484969d420188e9cff7348c") |
|
|
|
set(REQ_URL "https://gitee.com/mirrors/MKL-DNN/repository/archive/v1.6.tar.gz") |
|
|
|
set(MD5 "44da423a3b6848990a907f99a65b26e7") |
|
|
|
else() |
|
|
|
set(REQ_URL "https://github.com/oneapi-src/oneDNN/archive/v1.5.tar.gz") |
|
|
|
set(MD5 "5d97e0e8f4c0b37da5f524533b7a644b") |
|
|
|
set(REQ_URL "https://github.com/oneapi-src/oneDNN/archive/v1.6.tar.gz") |
|
|
|
set(MD5 "7cf251209f774ae6d61489ad7c2c3bea") |
|
|
|
endif () |
|
|
|
mindspore_add_pkg(onednn |
|
|
|
VER 1.5 |
|
|
|
VER 1.6 |
|
|
|
LIBS dnnl mkldnn |
|
|
|
URL ${REQ_URL} |
|
|
|
MD5 ${MD5} |
|
|
|
CMAKE_OPTION -DDNNL_ARCH_OPT_FLAGS='' -DDNNL_CPU_RUNTIME='SEQ' -DDNNL_BUILD_EXAMPLES=OFF -DDNNL_BUILD_TESTS=OFF) |
|
|
|
CMAKE_OPTION -DDNNL_ARCH_OPT_FLAGS='' -DDNNL_BUILD_EXAMPLES=OFF -DDNNL_BUILD_TESTS=OFF) |
|
|
|
endif() |
|
|
|
|
|
|
|
include_directories(${onednn_INC}) |
|
|
|
|