diff --git a/CMakeLists.txt b/CMakeLists.txt index b84c5a0032afd30ce2c135d910e76806b738d584..31d07d87cfaeb8e87db85c67215e1946c74720b3 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 3cab667e30db04aa874ccf99fe89a43d494affcc..f97cce513cbf10dd01f9805be088b5b91ec3488c 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