fix_pybind_op_reg_nokernel

8 years ago · 1540703394
parent 13b4749e24
commit 1540703394
1 changed files with 1 additions and 1 deletions
--- a/paddle/operators/CMakeLists.txt
+++ b/paddle/operators/CMakeLists.txt
@ -84,7 +84,7 @@ function(op_library TARGET)
    endif()

    # pybind USE_NO_KERNEL_OP
-    # operators may reuse code from
+    # HACK: if REGISTER_OP_CPU_KERNEL presents the operator must have kernel
    file(READ ${TARGET}.cc TARGET_CONTENT)
    string(REGEX MATCH "REGISTER_OP_CPU_KERNEL" regex_result "${TARGET_CONTENT}")
    string(REPLACE "_op" "" TARGET "${TARGET}")