CMakeLists.txt 1.3 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
set(PRIM_OP_PATH ${PADDLE_SOURCE_DIR}/paddle/fluid/operators/prim_ops/)

set(PRIM_OP_SRCS
    ${PRIM_OP_PATH}reshape_p_op.cc
    ${PRIM_OP_PATH}broadcast_p_op.cc
    ${PRIM_OP_PATH}reduce_sum_p_op.cc
    ${PRIM_OP_PATH}transpose_p_op.cc
    ${PRIM_OP_PATH}split_p_op.cc
    ${PRIM_OP_PATH}concat_p_op.cc
    ${PRIM_OP_PATH}slice_select_p_op.cc
    ${PRIM_OP_PATH}slice_assign_p_op.cc
    ${PRIM_OP_PATH}gather_p_op.cc
    ${PRIM_OP_PATH}scatter_add_p_op.cc
    ${PRIM_OP_PATH}add_p_op.cc
    ${PRIM_OP_PATH}sub_p_op.cc
    ${PRIM_OP_PATH}mul_p_op.cc
    ${PRIM_OP_PATH}div_p_op.cc
    ${PRIM_OP_PATH}sqrt_p_op.cc
    ${PRIM_OP_PATH}tanh_p_op.cc
    ${PRIM_OP_PATH}sin_p_op.cc
    ${PRIM_OP_PATH}cos_p_op.cc
    ${PRIM_OP_PATH}exp_p_op.cc
    ${PRIM_OP_PATH}matmul_p_op.cc
    ${PRIM_OP_PATH}fill_constant_p_op.cc
    ${PRIM_OP_PATH}log_p_op.cc
    ${PRIM_OP_PATH}select_p_op.cc
    ${PRIM_OP_PATH}eq_p_op.cc
    ${PRIM_OP_PATH}gt_p_op.cc
    ${PRIM_OP_PATH}ge_p_op.cc
    ${PRIM_OP_PATH}ne_p_op.cc
    ${PRIM_OP_PATH}pow_p_op.cc
    ${PRIM_OP_PATH}max_p_op.cc
    ${PRIM_OP_PATH}erf_p_op.cc
    ${PRIM_OP_PATH}bernoulli_p_op.cc
    ${PRIM_OP_PATH}abs_p_op.cc
    ${PRIM_OP_PATH}cast_p_op.cc
    ${PRIM_OP_PATH}rsqrt_p_op.cc
    ${PRIM_OP_PATH}uniform_random_p_op.cc)
cc_test_old(prim_op_test SRCS prim_op_test.cc ${PRIM_OP_SRCS} DEPS op_registry)