Skip to content

Commit f738691

Browse files
authored
Merge pull request #9740 from dzhwinter/memory/activation
"polish activation"
2 parents 8e00540 + ba5ddb7 commit f738691

File tree

7 files changed

+264
-386
lines changed

7 files changed

+264
-386
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)