提交 54fad183 编写于 作者: F fengjiayi

Fix error

上级 57c09784
......@@ -425,14 +425,15 @@ class OpKernelRegistrar : public Registrar {
__attribute__((unused)) = \
TouchOpKernelRegistrar_##op_type##_##DEVICE_TYPE()
#ifdef PADDLE_ONLY_CPU
#define USE_OP(op_type) \
#define USE_CPU_OP(op_type) \
USE_OP_ITSELF(op_type); \
USE_OP_KERNEL(op_type, CPU)
#ifdef PADDLE_ONLY_CPU
#define USE_OP(op_type) USE_CPU_OP(op_type)
#else
#define USE_OP(op_type) \
USE_OP_ITSELF(op_type); \
USE_OP_KERNEL(op_type, CPU); \
USE_CPU_OP(op_type); \
USE_OP_KERNEL(op_type, GPU)
#endif
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册