From f25d5b5629656f696d4357669158a4f2373d90b8 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 16 Sep 2020 14:03:41 +0800 Subject: [PATCH] fix(mge/build): fix version.ld path GitOrigin-RevId: 6d5bbf0d18df7cd2e0c9b8a42f84730da4873b50 --- imperative/src/version.ld | 1 + src/CMakeLists.txt | 6 +----- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/imperative/src/version.ld b/imperative/src/version.ld index 19c7cde5..876e7ddb 100644 --- a/imperative/src/version.ld +++ b/imperative/src/version.ld @@ -4,6 +4,7 @@ global: MEGDNN_VSYM_*; mgb_get_extern_c_opr_api_versioned; PyInit__imperative_rt; + PyInit__mgb; extern "C++" { *mgb::*; *megdnn::*; diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt index 733a8cee..d380d389 100644 --- a/src/CMakeLists.txt +++ b/src/CMakeLists.txt @@ -181,11 +181,7 @@ if(ANDROID) target_link_libraries(megbrain PUBLIC log) endif() -if(MGE_BUILD_IMPERATIVE_RT) - set (_VER_FILE ${PROJECT_SOURCE_DIR}/python_module/src/version.ld) -else() - set (_VER_FILE ${PROJECT_SOURCE_DIR}/imperative/src/version.ld) -endif() +set (_VER_FILE ${PROJECT_SOURCE_DIR}/imperative/src/version.ld) if(MGE_BUILD_IMPERATIVE_RT )