Skip to content

Commit 877e2ae

Browse files
committed
"fix ci"
1 parent e54f203 commit 877e2ae

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

paddle/fluid/operators/CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,12 @@ function(op_library TARGET)
163163

164164
# pybind USE_OP
165165
if (${pybind_flag} EQUAL 0)
166+
# NOTE(*): activation use macro to regist the kernels, set use_op manually.
167+
if(${TARGET} STREQUAL "activation")
168+
file(APPEND ${pybind_file} "USE_OP(relu);\n")
169+
else()
166170
file(APPEND ${pybind_file} "USE_OP(${TARGET});\n")
171+
endif()
167172
endif()
168173
endfunction()
169174

0 commit comments

Comments
 (0)