Closed
Description
Currently, we should manually add USE_OP(XXX)
in paddle/pybind.cc. To auto add USE_OP
, we plan to generate a pybind.h
file bypaddle/operators/CMakeLists.txt
. For different Ops, the rules are as follows:
net_op
doesn't need anyUSE_OP
.- if
XXX_op
only has.cc
file, it'sUSE_CPU_ONLY_OP
. - if
grep 'OperatorWithKernel' XXX_op.cc
is null, it'sUSE_NO_KERNEL_OP
USE_NO_KERNEL_OP
>USE_CPU_ONLY_OP
: meas that if a op is bothUSE_NO_KERNEL_OP
andUSE_CPU_ONLY_OP
, it isUSE_NO_KERNEL_OP
.
Metadata
Metadata
Assignees
Labels
No labels