CMakeLists.txt 1.0 KB
Newer Older
S
superjomn 已提交
1 2 3 4 5 6 7 8 9 10
#cc_library(runtime_lite SRCS runtime.cc)

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)


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

17
set(model_parser_deps variable_lite scope_lite ${tensor_lite} scope_lite
S
Superjomn 已提交
18 19
                      target_wrapper_host
                      compatible_pb_lite
Y
Yan Chunwei 已提交
20
                      memory_lite
S
Superjomn 已提交
21 22 23 24 25
                      )
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})
26 27

add_subdirectory(pb)