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

9 10 11
cc_library(
  conditional_block_op
  SRCS conditional_block_op.cc
R
risemeup1 已提交
12
  DEPS executor standalone_executor)
13 14 15 16 17 18 19
cc_library(
  op_variant
  SRCS op_variant.cc
  DEPS operator proto_desc)
cc_library(
  conditional_block_op_helper
  SRCS conditional_block_op_helper.cc
R
risemeup1 已提交
20
  DEPS op_variant operator conditional_block_op)
21 22 23
cc_library(
  recurrent_op_helper
  SRCS recurrent_op_helper.cc
R
risemeup1 已提交
24
  DEPS recurrent_op op_variant operator)
25 26 27
cc_library(
  while_op_helper
  SRCS while_op_helper.cc
R
risemeup1 已提交
28
  DEPS op_variant operator)
W
Wu Yi 已提交
29

30
if(WITH_UNITY_BUILD)
31
  target_link_libraries(paddle_operators_controlflow_unity conditional_block_op)
32
else()
33
  target_link_libraries(conditional_block_infer_op conditional_block_op)
34
endif()
Z
Zeng Jinle 已提交
35

36 37
file(APPEND ${pybind_file}
     "USE_OP_ITSELF(less_than);\nUSE_NO_KERNEL_OP(read_from_array);\n")
38 39 40 41
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"
)