|
|
|
@ -16,6 +16,12 @@ 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 MacOS" FORCE)
|
|
|
|
|
return()
|
|
|
|
|
ENDIF()
|
|
|
|
|
|
|
|
|
|
INCLUDE(ExternalProject)
|
|
|
|
|
SET(MKLML_DST_DIR "mklml")
|
|
|
|
|
SET(MKLML_INSTALL_ROOT "${THIRD_PARTY_PATH}/install")
|
|
|
|
@ -23,32 +29,24 @@ SET(MKLML_INSTALL_DIR ${MKLML_INSTALL_ROOT}/${MKLML_DST_DIR})
|
|
|
|
|
SET(MKLML_ROOT ${MKLML_INSTALL_DIR})
|
|
|
|
|
SET(MKLML_INC_DIR ${MKLML_ROOT}/include)
|
|
|
|
|
SET(MKLML_LIB_DIR ${MKLML_ROOT}/lib)
|
|
|
|
|
if(WIN32)
|
|
|
|
|
SET(CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_RPATH}" "${MKLML_ROOT}/lib")
|
|
|
|
|
|
|
|
|
|
SET(TIME_VERSION "2019.0.1.20181227")
|
|
|
|
|
IF(WIN32)
|
|
|
|
|
SET(MKLML_VER "mklml_win_${TIME_VERSION}" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_URL "https://paddlepaddledeps.cdn.bcebos.com/${MKLML_VER}.zip" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_LIB ${MKLML_LIB_DIR}/mklml.lib)
|
|
|
|
|
SET(MKLML_IOMP_LIB ${MKLML_LIB_DIR}/libiomp5md.lib)
|
|
|
|
|
SET(MKLML_SHARED_LIB ${MKLML_LIB_DIR}/mklml.dll)
|
|
|
|
|
SET(MKLML_SHARED_IOMP_LIB ${MKLML_LIB_DIR}/libiomp5md.dll)
|
|
|
|
|
else()
|
|
|
|
|
ELSE()
|
|
|
|
|
SET(MKLML_VER "mklml_lnx_${TIME_VERSION}" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_URL "http://paddlepaddledeps.cdn.bcebos.com/${MKLML_VER}.tgz" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_LIB ${MKLML_LIB_DIR}/libmklml_intel.so)
|
|
|
|
|
SET(MKLML_IOMP_LIB ${MKLML_LIB_DIR}/libiomp5.so)
|
|
|
|
|
SET(MKLML_SHARED_LIB ${MKLML_LIB_DIR}/libmklml_intel.so)
|
|
|
|
|
SET(MKLML_SHARED_IOMP_LIB ${MKLML_LIB_DIR}/libiomp5.so)
|
|
|
|
|
endif()
|
|
|
|
|
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.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.1.20180928" CACHE STRING "" FORCE)
|
|
|
|
|
SET(MKLML_URL "http://paddlepaddledeps.cdn.bcebos.com/${MKLML_VER}.tgz" CACHE STRING "" FORCE)
|
|
|
|
|
ENDIF()
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
SET(MKLML_PROJECT "extern_mklml")
|
|
|
|
|
MESSAGE(STATUS "MKLML_VER: ${MKLML_VER}, MKLML_URL: ${MKLML_URL}")
|
|
|
|
|