From 2c2df830513d7dc7f6fbb3b999c5ae51037bacaf Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 13 Jan 2022 18:15:05 +0800
Subject: [PATCH] fix(cmake): enable custom op when building develop to avoid
 the pytest fail

GitOrigin-RevId: fa05ead8996c0ee07916d34daea3aeef324d63ac
---
 CMakeLists.txt | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index e025b9b8e..22dbeaa3e 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -655,6 +655,10 @@ if(MGE_WITH_JIT_MLIR OR MGE_BUILD_IMPERATIVE_RT)
   include(cmake/llvm-project.cmake)
 endif()
 
+if(MGE_BUILD_IMPERATIVE_RT)
+  set(MGE_WITH_CUSTOM_OP ON)
+endif()
+
 if(MGE_WITH_DISTRIBUTED)
   include(cmake/protobuf.cmake)
   include(cmake/zmq.cmake)
-- 
GitLab