link iomp as needed

c4750414 · tensor-tang · 17c9e3d2 · c4750414
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

cmake/generic.cmake cmake/generic.cmake +1 -0

未找到文件。
--- a/cmake/generic.cmake
+++ b/cmake/generic.cmake
@@ -202,6 +202,7 @@ function(cc_library TARGET_NAME)
          list(APPEND cc_library_DEPS dynload_mklml)
        endif()
        add_dependencies(${TARGET_NAME} mklml)
+        target_link_libraries(${TARGET_NAME} "-L${MKLML_LIB_DIR} -liomp5 -Wl,--as-needed")
      endif()
      target_link_libraries(${TARGET_NAME} ${cc_library_DEPS})
      add_dependencies(${TARGET_NAME} ${cc_library_DEPS})