m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add macro USE_OP_GRADIENT()

Browse Source
revert-3824-remove_grad_op_type
fengjiayi 8 years ago
parent e4f058cec7
commit f66d78680d
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File

17
paddle/framework/op_registry.h
Unescape View File

@ -371,7 +371,7 @@ class OpKernelRegistrar : public Registrar {
static ::paddle::framework::GradOpRegistrar<grad_op_class> \
__op_gradient_registrar_##op_type##_##grad_op_type##__(#op_type, \
#grad_op_type); \
int TouchOpGradientRegister_##op_type() { \
int TouchOpGradientRegistrar_##op_type() { \
__op_gradient_registrar_##op_type##_##grad_op_type##__.Touch(); \
return 0; \
}
@ -416,6 +416,14 @@ class OpKernelRegistrar : public Registrar {
static int use_op_itself_##op_type##_ __attribute__((unused)) = \
TouchOpRegistrar_##op_type()
#define USE_OP_GRADIENT(op_type) \
STATIC_ASSERT_GLOBAL_NAMESPACE( \
__use_op_gradient_##op_type, \
"USE_OP_GRADIENT must be called in global namespace"); \
extern int TouchOpGradientRegistrar_##op_type(); \
static int use_op_gradient_##op_type##_ __attribute__((unused)) = \
TouchOpGradientRegistrar_##op_type()
#define USE_OP_KERNEL(op_type, DEVICE_TYPE) \
STATIC_ASSERT_GLOBAL_NAMESPACE( \
__use_op_kernel_##op_type##_##DEVICE_TYPE##__, \
@ -425,9 +433,10 @@ class OpKernelRegistrar : public Registrar {
__attribute__((unused)) = \
TouchOpKernelRegistrar_##op_type##_##DEVICE_TYPE()
#define USE_CPU_OP(op_type) \
USE_OP_ITSELF(op_type); \
USE_OP_KERNEL(op_type, CPU)
#define USE_CPU_OP(op_type) \
USE_OP_ITSELF(op_type); \
USE_OP_KERNEL(op_type, CPU); \
USE_OP_GRADIENT(op_type)
#ifdef PADDLE_ONLY_CPU
#define USE_OP(op_type) USE_CPU_OP(op_type)

Write Preview
Loading…
Cancel
Save