CMakeLists.txt 846 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
cc_library(
  jit_serializer
  SRCS serializer.cc
  DEPS lod_tensor device_context)

cc_library(
  jit_layer
  SRCS layer.cc
  DEPS executor parallel_executor executor_cache)

cc_library(
  jit_base_function
  SRCS base_function.cc
  DEPS scope proto_desc)

W
WangZhen 已提交
16 17 18
if(WITH_TESTING
   AND NOT WIN32
   AND NOT "$ENV{CI_SKIP_CPP_TEST}" STREQUAL "ON")
19 20
  add_custom_target(
    jit_download_program
W
WangZhen 已提交
21 22
    COMMAND wget -nc https://paddle-ci.gz.bcebos.com/dy2st/Testing.tar.gz
    COMMAND tar zxvf Testing.tar.gz)
23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
  set(JIT_DEPS
      phi
      elementwise_add_op
      matmul_v2_op
      activation_op
      reduce_mean_op
      feed_op
      fetch_op
      scale_op
      jit_serializer
      jit_layer
      jit_base_function)
  cc_test(
    layer_test
    SRCS layer_test.cc
    DEPS ${JIT_DEPS})
  add_dependencies(layer_test jit_download_program)
endif()