Browse Source

fix(cmake): enable custom op when building develop to avoid the pytest fail

GitOrigin-RevId: fa05ead899
tags/v1.8.0
Megvii Engine Team 3 years ago
parent
commit
2c2df83051
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      CMakeLists.txt

+ 4
- 0
CMakeLists.txt View File

@@ -655,6 +655,10 @@ if(MGE_WITH_JIT_MLIR OR MGE_BUILD_IMPERATIVE_RT)
include(cmake/llvm-project.cmake) include(cmake/llvm-project.cmake)
endif() endif()


if(MGE_BUILD_IMPERATIVE_RT)
set(MGE_WITH_CUSTOM_OP ON)
endif()

if(MGE_WITH_DISTRIBUTED) if(MGE_WITH_DISTRIBUTED)
include(cmake/protobuf.cmake) include(cmake/protobuf.cmake)
include(cmake/zmq.cmake) include(cmake/zmq.cmake)


Loading…
Cancel
Save