From ed92b9c1259af3ccc176b2d950df261952610c49 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 10 Jun 2022 16:00:19 +0800
Subject: [PATCH] fix(cmake): fix compilation options for benchmark in cmake

GitOrigin-RevId: d763fd32622b0ee9267ae04c637ff83639d90bb1
---
 CMakeLists.txt                 | 6 ++++++
 src/megbrain_build_config.h.in | 4 ++++
 2 files changed, 10 insertions(+)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index b84c5a00..31d07d87 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -88,6 +88,7 @@ option(MGE_ENABLE_LOGGING "Build with logging" ON)
 option(MGE_DEBUG_UTIL "Enable debug utility" ON)
 option(MGE_ENABLE_EXCEPTIONS "Build with exceptions" ON)
 option(MGE_WITH_TEST "Enable test for MegEngine." OFF)
+option(MGE_WITH_BENCHMARK "Enable DNN BENCHMARK" OFF)
 option(MGE_WITH_DISTRIBUTED "Build with distributed support" ON)
 option(MGE_BUILD_IMPERATIVE_RT "Build _imperative_rt Python Module " ON)
 option(MGE_INFERENCE_ONLY "Build inference only library." OFF)
@@ -1123,6 +1124,11 @@ if(MGE_WITH_TEST)
   set(MEGDNN_ENABLE_MULTI_THREADS 1)
 endif()
 
+# benchmark
+if(MGE_WITH_BENCHMARK)
+  set(MEGDNN_WITH_BENCHMARK ${MGE_WITH_BENCHMARK})
+endif()
+
 # CUDA
 set(MGB_CUDA ${MGE_WITH_CUDA})
 set(MEGDNN_WITH_CUDA ${MGE_WITH_CUDA})
diff --git a/src/megbrain_build_config.h.in b/src/megbrain_build_config.h.in
index 3cab667e..f97cce51 100644
--- a/src/megbrain_build_config.h.in
+++ b/src/megbrain_build_config.h.in
@@ -257,4 +257,8 @@
 #define MGE_WIN_DECLSPEC_DATA
 #endif
 
+#ifndef MEGDNN_WITH_BENCHMARK
+#define MEGDNN_WITH_BENCHMARK 0
+#endif
+
 #endif  // _HEADER_MGB_BUILD_CONFIG
-- 
GitLab