"fix ci"

7 years ago · 877e2aed03
parent e54f203c55
commit 877e2aed03
1 changed files with 5 additions and 0 deletions
--- a/paddle/fluid/operators/CMakeLists.txt
+++ b/paddle/fluid/operators/CMakeLists.txt
@ -163,8 +163,13 @@ function(op_library TARGET)

    # pybind USE_OP
    if (${pybind_flag} EQUAL 0)
+      # NOTE(*): activation use macro to regist the kernels, set use_op manually.
+      if(${TARGET} STREQUAL "activation")
+        file(APPEND ${pybind_file} "USE_OP(relu);\n")
+      else()
        file(APPEND ${pybind_file} "USE_OP(${TARGET});\n")
      endif()
+    endif()
 endfunction()

 add_subdirectory(math)