Fix error

54fad183 · fengjiayi · 57c09784 · 54fad183
隐藏空白更改
内联并排

Showing with 7 addition and 6 deletion

paddle/framework/op_registry.h paddle/framework/op_registry.h +7 -6

未找到文件。
--- a/paddle/framework/op_registry.h
+++ b/paddle/framework/op_registry.h
@@ -425,14 +425,15 @@ class OpKernelRegistrar : public Registrar {
      __attribute__((unused)) =                                  \
          TouchOpKernelRegistrar_##op_type##_##DEVICE_TYPE()

-#ifdef PADDLE_ONLY_CPU
-#define USE_OP(op_type)   \
-  USE_OP_ITSELF(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); \
+#define USE_OP(op_type) \
+  USE_CPU_OP(op_type);  \
  USE_OP_KERNEL(op_type, GPU)
 #endif