|
|
|
@ -16,14 +16,6 @@ IF(NOT ${WITH_MKLML})
|
|
|
|
|
return()
|
|
|
|
|
ENDIF(NOT ${WITH_MKLML})
|
|
|
|
|
|
|
|
|
|
IF(APPLE)
|
|
|
|
|
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 Windows and MacOS" FORCE)
|
|
|
|
|
return()
|
|
|
|
|
ENDIF()
|
|
|
|
|
|
|
|
|
|
INCLUDE(ExternalProject)
|
|
|
|
|
SET(MKLML_DST_DIR "mklml")
|
|
|
|
|
SET(MKLML_INSTALL_ROOT "${THIRD_PARTY_PATH}/install")
|
|
|
|
@ -47,10 +39,13 @@ SET(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_RPATH}" "${MKLML_ROOT}/lib")
|
|
|
|
|
IF((NOT DEFINED MKLML_VER) OR (NOT DEFINED MKLML_URL))
|
|
|
|
|
MESSAGE(STATUS "use pre defined download url")
|
|
|
|
|
if(WIN32)
|
|
|
|
|
SET(MKLML_VER "mklml_win_2019.0.20180710" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_VER "mklml_win_2019.0.1.20180928" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_URL "https://paddlepaddledeps.cdn.bcebos.com/${MKLML_VER}.zip" CACHE STRING "" FORCE)
|
|
|
|
|
elseif(APPLE)
|
|
|
|
|
SET(MKLML_VER "mklml_mac_2019.0.1.20180928" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_URL "http://paddlepaddledeps.cdn.bcebos.com/${MKLML_VER}.tgz" CACHE STRING "" FORCE)
|
|
|
|
|
else()
|
|
|
|
|
SET(MKLML_VER "mklml_lnx_2019.0.20180710" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_VER "mklml_lnx_2019.0.1.20180928" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_URL "http://paddlepaddledeps.cdn.bcebos.com/${MKLML_VER}.tgz" CACHE STRING "" FORCE)
|
|
|
|
|
ENDIF()
|
|
|
|
|
endif()
|
|
|
|
|