CMakeLists.txt 2.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
cc_test(
  test_copy_dev_api
  SRCS test_copy_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_dot_dev_api
  SRCS test_dot_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_creation_dev_api
  SRCS test_creation_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_flatten_dev_api
  SRCS test_flatten_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_matmul_dev_api
  SRCS test_matmul_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_mean_dev_api
  SRCS test_mean_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_scale_dev_api
  SRCS test_scale_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_cast_dev_api
  SRCS test_cast_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_elementwise_dev_api
  SRCS test_elementwise_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_reshape_dev_api
  SRCS test_reshape_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_sum_dev_api
  SRCS test_sum_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_conj_dev_api
  SRCS test_conj_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_concat_dev_api
  SRCS test_concat_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_split_dev_api
  SRCS test_split_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_sparse_utils_dev_api
  SRCS test_sparse_utils_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_sparse_conv3d_dev_api
  SRCS test_sparse_conv3d_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_sparse_pool_dev_api
  SRCS test_sparse_pool_dev_api.cc
  DEPS phi phi_api_utils)
cc_test(
  test_sparse_activation_dev_api
  SRCS test_sparse_activation_dev_api.cc
  DEPS phi phi_api_utils)
73 74 75 76
cc_test(
  test_sparse_elementwise_dev_api
  SRCS test_sparse_elementwise_dev_api.cc
  DEPS phi phi_api_utils)
77

78 79 80 81
cc_test(
  test_math_function
  SRCS test_math_function.cc
  DEPS math_function)
82
if(WITH_GPU)
83 84 85 86
  nv_test(
    test_math_function_gpu
    SRCS test_math_function.cu
    DEPS math_function)
87 88
endif()
if(WITH_ROCM)
89 90 91 92
  hip_test(
    test_math_function_gpu
    SRCS test_math_function.cu
    DEPS math_function)
93
endif()
F
Feiyu Chan 已提交
94

95 96 97 98
cc_test(
  test_cpu_vec
  SRCS test_cpu_vec.cc
  DEPS blas cpu_info)
J
Jack Zhou 已提交
99 100

# For String Kernels
101 102 103 104 105 106 107 108 109 110 111 112 113 114 115
cc_test(
  test_strings_lower_upper_dev_api
  SRCS test_strings_lower_upper_dev_api.cc
  DEPS phi phi_api_utils)
if(WITH_GPU)
  nv_test(
    test_strings_lower_upper_dev_gpu_api
    SRCS test_strings_lower_upper_dev_api.cu
    DEPS phi phi_api_utils)
elseif(WITH_ROCM)
  hip_test(
    test_strings_lower_upper_dev_gpu_api
    SRCS test_strings_lower_upper_dev_api.cu
    DEPS phi phi_api_utils)
endif()
J
Jack Zhou 已提交
116

117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
cc_test(
  test_strings_copy_dev_api
  SRCS test_strings_copy_dev_api.cc
  DEPS phi phi_api_utils)
if(WITH_GPU)
  nv_test(
    test_strings_copy_dev_gpu_api
    SRCS test_strings_copy_dev_api.cu
    DEPS phi phi_api_utils)
elseif(WITH_ROCM)
  hip_test(
    test_strings_copy_dev_gpu_api
    SRCS test_strings_copy_dev_api.cu
    DEPS phi phi_api_utils)
endif()