rm libmklml_gnu.so

7 years ago · d573195dde
parent 6332bd1ed8
commit d573195dde
1 changed files with 3 additions and 3 deletions
--- a/cmake/inference_lib.cmake
+++ b/cmake/inference_lib.cmake
@ -69,11 +69,11 @@ if(NOT CBLAS_FOUND)
      SRCS ${CBLAS_INSTALL_DIR}/lib ${CBLAS_INSTALL_DIR}/include
      DSTS ${dst_dir} ${dst_dir}
    )
-else()
+elseif (WITH_MKLML)
    set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/mklml")
    copy(mklml_lib
-      SRCS ${MKLML_LIB_DIR} ${MKLML_INC_DIR}
-      DSTS ${dst_dir} ${dst_dir}
+      SRCS ${MKLML_LIB} ${MKLML_IOMP_LIB} ${MKLML_INC_DIR}
+      DSTS ${dst_dir}/lib ${dst_dir}/lib ${dst_dir}
    )
 endif()