CMakeLists.txt 848 字节
Newer Older
S
Superjomn 已提交
1
set(cxx_api_lite_deps scope_lite host_kernels ops_lite optimizer_lite target_wrapper_host kernels_cuda optimizer_lite model_parser_lite)
S
superjomn 已提交
2
if(LITE_WITH_CUDA)
S
Superjomn 已提交
3 4
    cc_library(cxx_api_lite_cuda SRCS cxx_api.cc DEPS ${cxx_api_lite_deps} target_wrapper_cuda)
    nv_test(test_cxx_api_lite_cuda SRCS cxx_api_test.cc DEPS cxx_api_lite_cuda)
S
superjomn 已提交
5
endif()
S
Superjomn 已提交
6 7 8 9 10 11 12 13 14 15 16 17 18 19

cc_library(cxx_api_lite SRCS cxx_api.cc DEPS ${cxx_api_lite_deps})

set(light_api_deps
    scope_lite host_kernels ops_lite target_wrapper_host model_parser_lite)

if(LITE_WITH_CUDA)
    set(light_api_deps ${light_api_deps} target_wrapper_cuda)
endif()

cc_library(light_api_lite SRCS light_api.cc DEPS ${light_api_deps})

cc_test(test_cxx_api_lite SRCS cxx_api_test.cc DEPS cxx_api_lite model_parser_lite target_wrapper_host host_kernels)
cc_test(test_light_api SRCS light_api_test.cc DEPS light_api_lite)