CMakeLists.txt 785 字节
Newer Older
L
levi131 已提交
1 2
include(operators)
if(WITH_UNITY_BUILD)
3 4
  # Load Unity Build rules for operators in paddle/fluid/operators/prim_ops.
  include(unity_build_rule.cmake)
L
levi131 已提交
5 6 7
endif()
register_operators()

8
set(PRIM_OP_SRCS
L
levi131 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
    reshape_p_op.cc
    broadcast_p_op.cc
    reduce_p_op.cc
    transpose_p_op.cc
    split_p_op.cc
    concat_p_op.cc
    slice_select_p_op.cc
    slice_assign_p_op.cc
    gather_p_op.cc
    scatter_add_p_op.cc
    add_p_op.cc
    sub_p_op.cc
    mul_p_op.cc
    div_p_op.cc
    sqrt_p_op.cc
    tanh_p_op.cc
25 26 27
    sin_p_op.cc
    cos_p_op.cc
    exp_p_op.cc
L
levi131 已提交
28
    matmul_p_op.cc
29
    fill_constant_p_op.cc
30 31 32
    log_p_op.cc
    select_p_op.cc
    eq_p_op.cc
33
    pow_p_op.cc
34 35
    max_p_op.cc
    erf_p_op.cc)
L
levi131 已提交
36

37 38 39 40
cc_test(
  prim_op_test
  SRCS prim_op_test.cc ${PRIM_OP_SRCS}
  DEPS op_registry)