Skip to content

Commit 4b1a32d

Browse files
author
Yang Yang
committed
fix pybind.h generator
1 parent 411e888 commit 4b1a32d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/fluid/operators/CMakeLists.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -110,12 +110,12 @@ function(op_library TARGET)
110110
# Note that it's enough to just adding one operator to pybind in a *_op.cc file.
111111
# And for detail pybind information, please see generated paddle/pybind/pybind.h.
112112
file(READ ${TARGET}.cc TARGET_CONTENT)
113-
string(REGEX MATCH "REGISTER_OP\\(.*REGISTER_OP\\(" multi_register "${TARGET_CONTENT}")
114-
string(REGEX MATCH "REGISTER_OP\\([a-z0-9_]*," one_register "${multi_register}")
113+
string(REGEX MATCH "REGISTER_OPERATOR\\(.*REGISTER_OPERATOR\\(" multi_register "${TARGET_CONTENT}")
114+
string(REGEX MATCH "REGISTER_OPERATOR\\([a-z0-9_]*," one_register "${multi_register}")
115115
if (one_register STREQUAL "")
116116
string(REPLACE "_op" "" TARGET "${TARGET}")
117117
else ()
118-
string(REPLACE "REGISTER_OP(" "" TARGET "${one_register}")
118+
string(REPLACE "REGISTER_OPERATOR(" "" TARGET "${one_register}")
119119
string(REPLACE "," "" TARGET "${TARGET}")
120120
endif()
121121

0 commit comments

Comments
 (0)