CMakeLists.txt 2.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
if(LITE_WITH_ARM)
    lite_cc_test(test_transformer_with_mask_fp32_arm SRCS test_transformer_with_mask_fp32_arm.cc
      DEPS ${lite_model_test_DEPS} paddle_api_full
      ARM_DEPS ${arm_kernels}
      ARGS --model_dir=${LITE_MODEL_DIR}/transformer_with_mask_fp32 SERIAL)
  if(WITH_TESTING)
      add_dependencies(test_transformer_with_mask_fp32_arm extern_lite_download_transformer_with_mask_fp32_tar_gz)
  endif()
endif()

11 12
if(LITE_WITH_XPU AND NOT LITE_WITH_XTCL)
    lite_cc_test(test_resnet50_fp32_xpu SRCS test_resnet50_fp32_xpu.cc
13 14 15
      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)
16
    lite_cc_test(test_ernie_fp32_xpu SRCS test_ernie_fp32_xpu.cc
17 18
      DEPS mir_passes lite_api_test_helper paddle_api_full paddle_api_light gflags utils
      ${ops} ${host_kernels} ${x86_kernels} ${xpu_kernels}
H
hong19860320 已提交
19
      ARGS --model_dir=${LITE_MODEL_DIR}/ernie)
20
    lite_cc_test(test_bert_fp32_xpu SRCS test_bert_fp32_xpu.cc
21 22
      DEPS mir_passes lite_api_test_helper paddle_api_full paddle_api_light gflags utils
      ${ops} ${host_kernels} ${x86_kernels} ${xpu_kernels}
H
hong19860320 已提交
23
      ARGS --model_dir=${LITE_MODEL_DIR}/bert)
24
    if(WITH_TESTING)
25 26 27
        add_dependencies(test_resnet50_fp32_xpu extern_lite_download_resnet50_tar_gz)
        add_dependencies(test_ernie_fp32_xpu extern_lite_download_ernie_tar_gz)
        add_dependencies(test_bert_fp32_xpu extern_lite_download_bert_tar_gz)
28
    endif()
29
    # TODO(miaotianxiang): enable later
30
    #lite_cc_test(test_fpr_fp32_xpu SRCS test_fpr_fp32_xpu.cc
31 32 33
      #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)
34
    #lite_cc_test(test_mmdnn_fp32_xpu SRCS test_mmdnn_fp32_xpu.cc
35 36 37
      #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)
38
endif()
39 40 41 42 43 44 45

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 已提交
46 47 48 49 50 51 52

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()