add OpenMP to top cmake

intel · Dec 29, 2023 · 5fe0d8e · 5fe0d8e
1 parent 76e8b52
commit 5fe0d8e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -143,6 +143,8 @@ endif()
 if (NE_BTLA_UT)
   set(BTLA_UT_ALL ON)
 endif()
+include(FindOpenMP)
+
 add_subdirectory(bestla)
 
 add_subdirectory(neural_speed)
diff --git a/neural_speed/cmake/Common.cmake b/neural_speed/cmake/Common.cmake
@@ -31,13 +31,15 @@ endfunction()
 
 function(add_executable_w_warning TARGET)
     add_executable(${TARGET} ${ARGN})
+    target_link_libraries(${TARGET} PUBLIC bestla::bestla OpenMP::OpenMP_CXX OpenMP::OpenMP_C)
     set_target_properties(${TARGET} PROPERTIES C_STANDARD 11 C_STANDARD_REQUIRED ON C_EXTENSIONS OFF)
     set_target_properties(${TARGET} PROPERTIES CXX_STANDARD 11 CXX_STANDARD_REQUIRED ON CXX_EXTENSIONS OFF)
     warning_check(${TARGET})
 endfunction()
 
 function(add_library_w_warning_ TARGET)
     add_library(${TARGET} ${ARGN})
+    target_link_libraries(${TARGET} PUBLIC bestla::bestla OpenMP::OpenMP_CXX OpenMP::OpenMP_C)
     set_target_properties(${TARGET} PROPERTIES C_STANDARD 11 C_STANDARD_REQUIRED ON C_EXTENSIONS OFF)
     set_target_properties(${TARGET} PROPERTIES CXX_STANDARD 11 CXX_STANDARD_REQUIRED ON CXX_EXTENSIONS OFF)
     warning_check(${TARGET})