CMakeLists.txt 1.3 KB
Newer Older
W
Wu Yi 已提交
1
include(operators)
2 3 4 5
if(WITH_UNITY_BUILD)
    # Load Unity Build rules for operators in paddle/fluid/operators/controlflow.
    include(unity_build_rule.cmake)
endif()
Z
zhang wenhui 已提交
6 7 8
register_operators(EXCLUDES conditional_block_op DEPS naive_executor)

cc_library(conditional_block_op SRCS conditional_block_op.cc DEPS executor)
9
cc_library(op_variant SRCS op_variant.cc DEPS operator proto_desc)
Z
Zeng Jinle 已提交
10
cc_library(conditional_block_op_helper SRCS conditional_block_op_helper.cc DEPS operator op_variant conditional_block_op)
11 12
cc_library(recurrent_op_helper SRCS recurrent_op_helper.cc DEPS operator op_variant recurrent_op)
cc_library(while_op_helper SRCS while_op_helper.cc DEPS operator op_variant) 
W
Wu Yi 已提交
13

14 15
cc_test(conditional_block_op_test SRCS conditional_block_op_test.cc DEPS conditional_block_op executor)

16 17 18 19 20
if(WITH_UNITY_BUILD)
    target_link_libraries(paddle_operators_controlflow_unity conditional_block_op)
else()
    target_link_libraries(conditional_block_infer_op conditional_block_op)
endif()
Z
Zeng Jinle 已提交
21

F
From00 已提交
22
file(APPEND ${pybind_file} "USE_OP_ITSELF(less_than);\nUSE_OP_ITSELF(equal_all);\nUSE_NO_KERNEL_OP(read_from_array);\n")
23
file(APPEND ${pybind_file} "USE_OP_ITSELF(logical_and);\nUSE_OP_ITSELF(logical_or);\nUSE_OP_ITSELF(logical_xor);\nUSE_OP_ITSELF(logical_not);\n")
24
file(APPEND ${pybind_file} "USE_OP(bitwise_and);\nUSE_OP(bitwise_or);\nUSE_OP(bitwise_xor);\nUSE_OP(bitwise_not);\n")