CMakeLists.txt 997 字节
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 20 21 22 23 24
                      target_wrapper_host
                      compatible_pb_lite
                      )
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})
25 26

add_subdirectory(pb)
27