CMakeLists.txt

include(operators)
if(WITH_UNITY_BUILD)
    # Load Unity Build rules for operators in paddle/fluid/operators/prim_ops.
    include(unity_build_rule.cmake)
endif()
register_operators()

SET(PRIM_OP_SRCS
    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
    matmul_p_op.cc
    fill_constant_p_op.cc)

cc_test(prim_op_test SRCS prim_op_test.cc ${PRIM_OP_SRCS} DEPS op_registry)