File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -110,12 +110,12 @@ function(op_library TARGET)
110
110
# Note that it's enough to just adding one operator to pybind in a *_op.cc file.
111
111
# And for detail pybind information, please see generated paddle/pybind/pybind.h.
112
112
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} " )
115
115
if (one_register STREQUAL "" )
116
116
string (REPLACE "_op" "" TARGET "${TARGET} " )
117
117
else ()
118
- string (REPLACE "REGISTER_OP (" "" TARGET "${one_register} " )
118
+ string (REPLACE "REGISTER_OPERATOR (" "" TARGET "${one_register} " )
119
119
string (REPLACE "," "" TARGET "${TARGET} " )
120
120
endif ()
121
121
You can’t perform that action at this time.
0 commit comments