From a6d5db83b0a01e45f77732bfe67ced306d2dce9c Mon Sep 17 00:00:00 2001
From: vsooda <liushouda@qq.com>
Date: Wed, 6 Dec 2017 10:01:56 +0800
Subject: [PATCH] fix openmp not found

---
 CMakeLists.txt | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 0968e75f8..ce68f7a0e 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -27,8 +27,10 @@ option(NCNN_OPENCV "minimal opencv structure emulation" OFF)
 
 if(NCNN_OPENMP)
     find_package(OpenMP)
-    set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
-    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
+    if(OpenMP_CXX_FOUND OR OPENMP_FOUND)
+        set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
+        set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
+    endif()
 endif()
 
 add_definitions(-Wall -Wextra)