|
|
|
@ -75,25 +75,26 @@ copy(inference_lib_dist
|
|
|
|
|
SRCS ${DLPACK_INCLUDE_DIR}/dlpack
|
|
|
|
|
DSTS ${dst_dir})
|
|
|
|
|
|
|
|
|
|
if(WITH_MKLML)
|
|
|
|
|
|
|
|
|
|
if(${CBLAS_PROVIDER} STREQUAL MKLML)
|
|
|
|
|
set(dst_dir "${FLUID_INFERENCE_INSTALL_DIR}/third_party/install/mklml")
|
|
|
|
|
if(WIN32)
|
|
|
|
|
copy(inference_lib_dist
|
|
|
|
|
SRCS ${MKLML_LIB} ${MKLML_IOMP_LIB} ${MKLML_SHARED_LIB}
|
|
|
|
|
${MKLML_SHARED_LIB_DEPS} ${MKLML_SHARED_IOMP_LIB} ${MKLML_INC_DIR}
|
|
|
|
|
${MKLML_SHARED_LIB_DEPS} ${MKLML_SHARED_IOMP_LIB} ${MKLML_INC_DIR}
|
|
|
|
|
DSTS ${dst_dir}/lib ${dst_dir}/lib ${dst_dir}/lib
|
|
|
|
|
${dst_dir}/lib ${dst_dir}/lib ${dst_dir})
|
|
|
|
|
${dst_dir}/lib ${dst_dir}/lib ${dst_dir})
|
|
|
|
|
else()
|
|
|
|
|
copy(inference_lib_dist
|
|
|
|
|
SRCS ${MKLML_LIB} ${MKLML_IOMP_LIB} ${MKLML_INC_DIR}
|
|
|
|
|
DSTS ${dst_dir}/lib ${dst_dir}/lib ${dst_dir})
|
|
|
|
|
endif()
|
|
|
|
|
elseif (NOT CBLAS_FOUND OR WIN32)
|
|
|
|
|
elseif(${CBLAS_PROVIDER} STREQUAL EXTERN_OPENBLAS)
|
|
|
|
|
set(dst_dir "${FLUID_INFERENCE_INSTALL_DIR}/third_party/install/openblas")
|
|
|
|
|
copy(inference_lib_dist
|
|
|
|
|
SRCS ${CBLAS_INSTALL_DIR}/lib ${CBLAS_INSTALL_DIR}/include
|
|
|
|
|
DSTS ${dst_dir} ${dst_dir})
|
|
|
|
|
endif ()
|
|
|
|
|
copy(inference_lib_dist
|
|
|
|
|
SRCS ${CBLAS_INSTALL_DIR}/lib ${CBLAS_INSTALL_DIR}/include
|
|
|
|
|
DSTS ${dst_dir} ${dst_dir})
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
if(WITH_MKLDNN)
|
|
|
|
|
set(dst_dir "${FLUID_INFERENCE_INSTALL_DIR}/third_party/install/mkldnn")
|
|
|
|
|