diff --git a/CMakeLists.txt b/CMakeLists.txt
index e025b9b8e0b04d3762abd717dd859c4f5d4d5242..22dbeaa3e1d54b26ab3e75a1d5f6a957ad3b7878 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)