提交 394edd86 编写于 作者: Z zhouwei25 提交者: Jiabin Yang

fix mklml and cblas bug,test=develop (#20970)

上级 de9bec60
...@@ -19,16 +19,6 @@ set(THIRD_PARTY_PATH "${CMAKE_BINARY_DIR}/third_party" CACHE STRING ...@@ -19,16 +19,6 @@ set(THIRD_PARTY_PATH "${CMAKE_BINARY_DIR}/third_party" CACHE STRING
set(THIRD_PARTY_BUILD_TYPE Release) set(THIRD_PARTY_BUILD_TYPE Release)
set(WITH_MKLML ${WITH_MKL})
if (NOT DEFINED WITH_MKLDNN)
if (WITH_MKL AND AVX2_FOUND)
set(WITH_MKLDNN ON)
else()
message(STATUS "Do not have AVX2 intrinsics and disabled MKL-DNN")
set(WITH_MKLDNN OFF)
endif()
endif()
# Correction of flags on different Platform(WIN/MAC) and Print Warning Message # Correction of flags on different Platform(WIN/MAC) and Print Warning Message
if (APPLE) if (APPLE)
if(WITH_MKL) if(WITH_MKL)
...@@ -36,12 +26,6 @@ if (APPLE) ...@@ -36,12 +26,6 @@ if (APPLE)
"Mac is not supported with MKL in Paddle yet. Force WITH_MKL=OFF.") "Mac is not supported with MKL in Paddle yet. Force WITH_MKL=OFF.")
set(WITH_MKL OFF CACHE STRING "Disable MKL for building on mac" FORCE) set(WITH_MKL OFF CACHE STRING "Disable MKL for building on mac" FORCE)
endif() endif()
if(WITH_MKLML)
MESSAGE(WARNING
"Mac is not supported with MKLML in Paddle yet. Force WITH_MKLML=OFF.")
set(WITH_MKLML OFF CACHE STRING "Disable MKLML package in MacOS" FORCE)
endif()
endif() endif()
if(WIN32 OR APPLE) if(WIN32 OR APPLE)
...@@ -67,6 +51,16 @@ if(WIN32 OR APPLE) ...@@ -67,6 +51,16 @@ if(WIN32 OR APPLE)
endif() endif()
endif() endif()
set(WITH_MKLML ${WITH_MKL})
if (NOT DEFINED WITH_MKLDNN)
if (WITH_MKL AND AVX2_FOUND)
set(WITH_MKLDNN ON)
else()
message(STATUS "Do not have AVX2 intrinsics and disabled MKL-DNN")
set(WITH_MKLDNN OFF)
endif()
endif()
if(WIN32 OR APPLE OR NOT WITH_GPU OR ON_INFER) if(WIN32 OR APPLE OR NOT WITH_GPU OR ON_INFER)
set(WITH_DGC OFF) set(WITH_DGC OFF)
endif() endif()
...@@ -93,10 +87,11 @@ endif() ...@@ -93,10 +87,11 @@ endif()
if(WITH_MKLML) if(WITH_MKLML)
include(external/mklml) # download, install mklml package include(external/mklml) # download, install mklml package
list(APPEND third_party_deps mklml) list(APPEND third_party_deps mklml)
elseif (NOT CBLAS_FOUND OR WIN32)
list(APPEND third_party_deps extern_openblas)
endif() endif()
include(external/openblas) # find first, then download, build, install openblas include(external/openblas) # find first, then download, build, install openblas
if(NOT CBLAS_FOUND)
list(APPEND third_party_deps extern_openblas)
endif()
if(WITH_MKLDNN) if(WITH_MKLDNN)
include(external/mkldnn) # download, build, install mkldnn include(external/mkldnn) # download, build, install mkldnn
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册