only link iomp when with MKLDNN and MKLML

f96d31d5 · tensor-tang · 0f7c4071 · f96d31d5
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

cmake/util.cmake cmake/util.cmake +4 -0

未找到文件。
--- a/cmake/util.cmake
+++ b/cmake/util.cmake
@@ -97,6 +97,10 @@ function(link_paddle_exe TARGET_NAME)
        target_link_libraries(${TARGET_NAME} log)
    endif(ANDROID)

+    if(WITH_MKLDNN AND WITH_MKLML AND MKLDNN_IOMP_DIR)
+      target_link_libraries(${TARGET_NAME} "-L${MKLDNN_IOMP_DIR} -liomp5 -Wl,--as-needed")
+    endif()
+
    add_dependencies(${TARGET_NAME} ${external_project_dependencies})
 endfunction()