提交 5974c1b7 编写于 作者: L Luo Tao

refine comments in CMakelists.txt of operator

上级 e4e95bee
...@@ -61,7 +61,7 @@ function(op_library TARGET) ...@@ -61,7 +61,7 @@ function(op_library TARGET)
${op_common_deps}) ${op_common_deps})
endif() endif()
# net_op doesn't need pybind, others will be pybind manually # Define operators that don't need pybind here.
foreach(manual_pybind_op "net_op" "compare_op" "logical_op" "nccl_op" "tensor_array_read_write_op") foreach(manual_pybind_op "net_op" "compare_op" "logical_op" "nccl_op" "tensor_array_read_write_op")
if ("${TARGET}" STREQUAL "${manual_pybind_op}") if ("${TARGET}" STREQUAL "${manual_pybind_op}")
set(pybind_flag 1) set(pybind_flag 1)
...@@ -69,7 +69,8 @@ function(op_library TARGET) ...@@ -69,7 +69,8 @@ function(op_library TARGET)
endforeach() endforeach()
file(READ ${TARGET}.cc TARGET_CONTENT) file(READ ${TARGET}.cc TARGET_CONTENT)
# It's enough to just adding one operator to pybind # It's enough to just adding one operator to pybind.
# And for detail pybind information, please see paddle/pybind/pybind.h.
string(REGEX MATCH "REGISTER_OP\\(.*REGISTER_OP\\(" multi_register "${TARGET_CONTENT}") string(REGEX MATCH "REGISTER_OP\\(.*REGISTER_OP\\(" multi_register "${TARGET_CONTENT}")
string(REGEX MATCH "REGISTER_OP\\([a-z0-9_]*," one_register "${multi_register}") string(REGEX MATCH "REGISTER_OP\\([a-z0-9_]*," one_register "${multi_register}")
if (one_register STREQUAL "") if (one_register STREQUAL "")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册