CMakeLists.txt

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)
if(WITH_CINN)
  set(CINN_DEPS cinn_compiler)
endif()
cc_test_old(
  prim_op_test
  SRCS
  prim_op_test.cc
  ${PRIM_OP_SRCS}
  DEPS
  fleet_executor
  ${CINN_DEPS}
  op_registry
  python)