From 54fad18382741baa5b7965130a215daa137aa03d Mon Sep 17 00:00:00 2001 From: fengjiayi Date: Wed, 9 Aug 2017 14:03:37 -0700 Subject: [PATCH] Fix error --- paddle/framework/op_registry.h | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/paddle/framework/op_registry.h b/paddle/framework/op_registry.h index 3858cd7794d..3633ddb9df2 100644 --- 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 -- GitLab