CMakeLists.txt 1.1 KB
Newer Older
S
superjomn 已提交
1 2
#cc_library(runtime_lite SRCS runtime.cc)

Y
Yan Chunwei 已提交
3 4 5 6 7 8 9 10 11
#TODO(Superjomn) enable it again.
if(NOT LITE_WITH_LIGHT_WEIGHT_FRAMEWORK)
    lite_cc_test(test_model_parser_lite SRCS model_parser_test.cc
      DEPS model_parser_lite framework_proto_lite
      ARGS --model_dir=${LITE_MODEL_DIR}/lite_naive_model)
    if(WITH_TESTING)
        add_dependencies(test_model_parser_lite extern_lite_download_lite_naive_model_tar_gz)
    endif(WITH_TESTING)
endif()
S
superjomn 已提交
12 13


14
if(LITE_WITH_LIGHT_WEIGHT_FRAMEWORK)
15
    cc_library(compatible_pb_lite SRCS compatible_pb.cc DEPS op_desc_lite framework_proto_lite var_desc_lite)
16
else()
S
Superjomn 已提交
17
    cc_library(compatible_pb_lite SRCS compatible_pb.cc DEPS framework_proto_lite proto_desc)
18 19
endif(LITE_WITH_LIGHT_WEIGHT_FRAMEWORK)

20
set(model_parser_deps variable_lite scope_lite ${tensor_lite} scope_lite
S
Superjomn 已提交
21 22
                      target_wrapper_host
                      compatible_pb_lite
Y
Yan Chunwei 已提交
23
                      memory_lite
S
Superjomn 已提交
24 25 26 27 28
                      )
if (LITE_WITH_CUDA)
    set(model_parser_deps ${model_parser_deps} target_wrapper_cuda)
endif()
cc_library(model_parser_lite SRCS model_parser.cc DEPS ${model_parser_deps})
29 30

add_subdirectory(pb)