CMakeLists.txt 1.6 KB
Newer Older
1 2 3 4 5 6 7 8
if(LITE_WITH_XPU)
    lite_cc_test(test_resnet50_lite_xpu SRCS test_resnet50_lite_xpu.cc
      DEPS mir_passes lite_api_test_helper paddle_api_full paddle_api_light gflags utils
      ${ops} ${host_kernels} ${x86_kernels} ${xpu_kernels}
      ARGS --model_dir=${LITE_MODEL_DIR}/resnet50)
    lite_cc_test(test_ernie_lite_xpu SRCS test_ernie_lite_xpu.cc
      DEPS mir_passes lite_api_test_helper paddle_api_full paddle_api_light gflags utils
      ${ops} ${host_kernels} ${x86_kernels} ${xpu_kernels}
H
hong19860320 已提交
9
      ARGS --model_dir=${LITE_MODEL_DIR}/ernie)
10 11 12
    lite_cc_test(test_bert_lite_xpu SRCS test_bert_lite_xpu.cc
      DEPS mir_passes lite_api_test_helper paddle_api_full paddle_api_light gflags utils
      ${ops} ${host_kernels} ${x86_kernels} ${xpu_kernels}
H
hong19860320 已提交
13
      ARGS --model_dir=${LITE_MODEL_DIR}/bert)
14 15 16 17 18
    if(WITH_TESTING)
        add_dependencies(test_resnet50_lite_xpu extern_lite_download_resnet50_tar_gz)
        add_dependencies(test_ernie_lite_xpu extern_lite_download_ernie_tar_gz)
        add_dependencies(test_bert_lite_xpu extern_lite_download_bert_tar_gz)
    endif()
19
endif()
20 21 22 23 24 25 26

if(LITE_WITH_RKNPU)
    lite_cc_test(test_mobilenetv1_int8_rknpu SRCS test_mobilenetv1_int8_rknpu.cc
      DEPS ${lite_model_test_DEPS} paddle_api_full
      RKNPU_DEPS ${rknpu_kernels} ${rknpu_bridges}
      ARGS --model_dir=${LITE_MODEL_DIR}/MobilenetV1_full_quant SERIAL)
endif()
H
hong19860320 已提交
27 28 29 30 31 32 33

if(LITE_WITH_APU)
    lite_cc_test(test_mobilenetv1_int8_apu SRCS test_mobilenetv1_int8_apu.cc
      DEPS ${lite_model_test_DEPS} paddle_api_full
      APU_DEPS ${apu_kernels} ${apu_bridges}
      ARGS --model_dir=${LITE_MODEL_DIR}/MobilenetV1_full_quant SERIAL)
endif()