CMakeLists.txt 2.0 KB
Newer Older
Y
Yan Chunwei 已提交
1 2 3
# Generate paddle_lite_jni.so

if (LITE_ON_TINY_PUBLISH)
4 5
    set(CMAKE_CXX_FLAGS_RELEASE "-Os -DNDEBUG")
    set(CMAKE_C_FLAGS_RELEASE "-Os -DNDEBUG")
Y
Yan Chunwei 已提交
6 7 8 9 10 11 12 13 14 15 16 17 18 19
    set(lib_DEPS light_api paddle_api paddle_api_light)
else()
    set(lib_DEPS light_api cxx_api paddle_api_full paddle_api paddle_api_light)
endif()

include_directories(${JNI_INCLUDE_DIRS} ${_classDir} ${_stubDir})
if (NOT LITE_ON_TINY_PUBLISH)
    lite_cc_library(paddle_lite_jni MODULE 
        SRCS paddle_lite_jni.cc tensor_jni.cc
        DEPS ${lib_DEPS}
        ARM_DEPS ${arm_kernels} NPU_DEPS ${npu_kernels})
    # Unlike static library, module library has to link target to be able to work
    # as a single .so lib.
    target_link_libraries(paddle_lite_jni ${lib_DEPS} ${arm_kernels} ${npu_kernels})
20 21 22 23 24 25
    if (LITE_WITH_NPU)
        # Strips the symbols of our protobuf functions to fix the conflicts during
        # loading HIAI builder libs (libhiai_ir.so and libhiai_ir_build.so)
        set(LINK_FLAGS "-Wl,--version-script ${PADDLE_SOURCE_DIR}/lite/core/lite.map")
        set_target_properties(paddle_lite_jni PROPERTIES LINK_FLAGS "${LINK_FLAGS}")
    endif()
Y
Yan Chunwei 已提交
26 27
else()
    add_library(paddle_lite_jni SHARED "")
28 29 30
    set(TARGET_COMIPILE_FLAGS "-fdata-sections")
    if (NOT (ARM_TARGET_LANG STREQUAL "clang")) #gcc
        set(TARGET_COMIPILE_FLAGS "${TARGET_COMIPILE_FLAGS} -flto")
31
    endif()
32
    set_target_properties(paddle_lite_jni PROPERTIES COMPILE_FLAGS ${TARGET_COMIPILE_FLAGS})
Y
Yan Chunwei 已提交
33
    target_sources(paddle_lite_jni PUBLIC ${__lite_cc_files} paddle_lite_jni.cc tensor_jni.cc)
S
sangoly 已提交
34
    add_dependencies(paddle_lite_jni op_list_h kernel_list_h)
35 36
    if (LITE_WITH_NPU)
        # Need to add HIAI runtime libs (libhiai.so) dependency
37
        target_link_libraries(paddle_lite_jni ${npu_builder_libs} ${npu_runtime_libs})
38
    endif()
Y
Yan Chunwei 已提交
39 40 41 42 43 44 45 46 47
endif()

if (APPLE)
    # MacOS only accepts JNI lib ends with .jnilib or .dylib
    set_target_properties(paddle_lite_jni PROPERTIES SUFFIX ".jnilib")
elseif (WIN32)
    # Windows only accepts JNI lib ends with .dll
    set_target_properties(paddle_lite_jni PROPERTIES SUFFIX ".dll")
endif (APPLE)