提交 15407033 编写于 作者: T typhoonzero

fix_pybind_op_reg_nokernel

上级 13b4749e
......@@ -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}")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册