CMakeLists.txt 1.7 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 32 33 34 35 36 37 38 39
cc_test(
  test_conv_mkldnn_nhwc
  SRCS test_conv_mkldnn_nhwc.cc
  DEPS executor
       op_registry
       depthwise_conv
       tensor
       phi
       scope
       device_context
       enforce
       generated_static_op)

40 41 42 43 44
set(TEST_MKLDNN_CACHING_DEPS
    op_registry
    elementwise_mul_op
    elementwise_add_op
    activation_op
45
    phi
46 47
    scope
    device_context
48 49 50
    enforce
    generated_static_op)

51 52 53 54 55 56 57
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})
58
if(WIN32 AND WITH_TESTING)
R
risemeup1 已提交
59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
  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)
82 83 84
else()
  cc_test_old(test_mkldnn_op_nhwc SRCS test_mkldnn_op_nhwc.cc DEPS
              ${paddle_lib} python)
R
risemeup1 已提交
85
endif()