CMakeLists.txt 1.5 KB
Newer Older
1
get_property(paddle_lib GLOBAL PROPERTY PADDLE_LIB_NAME)
2 3 4
cc_test(
  test_mkldnn_op_inplace
  SRCS test_mkldnn_op_inplace.cc
R
risemeup1 已提交
5 6
  DEPS executor
       op_registry
7 8
       elementwise_add_op
       activation_op
9
       phi
10 11 12
       scope
       device_context
       enforce
13
       generated_static_op)
14 15 16 17 18 19 20 21 22 23 24 25
cc_test(
  test_mkldnn_cpu_quantize_pass
  SRCS test_mkldnn_cpu_quantize_pass.cc
  DEPS executor
       op_registry
       activation_op
       conv_activation_mkldnn_fuse_pass
       cpu_quantize_placement_pass
       cpu_quantize_pass
       phi
       scope
       device_context)
26 27 28 29 30 31

set(TEST_MKLDNN_CACHING_DEPS
    op_registry
    elementwise_mul_op
    elementwise_add_op
    activation_op
32
    phi
33 34
    scope
    device_context
35 36 37
    enforce
    generated_static_op)

38 39 40 41 42 43 44
if(WITH_GPU OR WITH_ROCM)
  set(TEST_MKLDNN_CACHING_DEPS ${TEST_MKLDNN_CACHING_DEPS} depthwise_conv)
endif()
cc_test(
  test_mkldnn_caching
  SRCS test_mkldnn_caching.cc
  DEPS ${TEST_MKLDNN_CACHING_DEPS})
45
if(WIN32 AND WITH_TESTING)
R
risemeup1 已提交
46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
  cc_test_old(
    test_mkldnn_op_nhwc
    SRCS
    test_mkldnn_op_nhwc.cc
    DEPS
    fleet_executor
    conditional_block_op
    standalone_executor
    executor
    recurrent_op_helper
    recurrent_op
    op_registry
    generated_static_op
    crop_op
    activation_op
    generated_op
    generated_static_op
    phi
    transpose_op
    fused_transpose_op
    scope
    device_context
    enforce)
69 70 71
else()
  cc_test_old(test_mkldnn_op_nhwc SRCS test_mkldnn_op_nhwc.cc DEPS
              ${paddle_lib} python)
R
risemeup1 已提交
72
endif()