@@ -177,16 +177,16 @@ class OpKernelRegistrar : public Registrar {
177
177
/* *
178
178
* Macro to register OperatorKernel.
179
179
*/
180
- #define REGISTER_OP_KERNEL (op_type, DEVICE_TYPE , place_class, ...) \
181
- STATIC_ASSERT_GLOBAL_NAMESPACE ( \
182
- __reg_op_kernel_##op_type##_##DEVICE_TYPE ##__, \
183
- " REGISTER_OP_KERNEL must be called in global namespace" ); \
184
- static ::paddle::framework::OpKernelRegistrar<place_class, __VA_ARGS__> \
185
- __op_kernel_registrar_##op_type##_##DEVICE_TYPE ##__(#op_type, \
186
- #DEVICE_TYPE); \
187
- int TouchOpKernelRegistrar_##op_type##_##DEVICE_TYPE () { \
188
- __op_kernel_registrar_##op_type##_##DEVICE_TYPE ##__.Touch (); \
189
- return 0 ; \
180
+ #define REGISTER_OP_KERNEL (op_type, LIBRARY_TYPE , place_class, ...) \
181
+ STATIC_ASSERT_GLOBAL_NAMESPACE ( \
182
+ __reg_op_kernel_##op_type##_##LIBRARY_TYPE ##__, \
183
+ " REGISTER_OP_KERNEL must be called in global namespace" ); \
184
+ static ::paddle::framework::OpKernelRegistrar<place_class, __VA_ARGS__> \
185
+ __op_kernel_registrar_##op_type##_##LIBRARY_TYPE ##__(#op_type, \
186
+ #LIBRARY_TYPE); \
187
+ int TouchOpKernelRegistrar_##op_type##_##LIBRARY_TYPE () { \
188
+ __op_kernel_registrar_##op_type##_##LIBRARY_TYPE ##__.Touch (); \
189
+ return 0 ; \
190
190
}
191
191
192
192
#define REGISTER_OP_CUDA_KERNEL (op_type, ...) \
@@ -208,14 +208,14 @@ class OpKernelRegistrar : public Registrar {
208
208
static int use_op_itself_##op_type##_ __attribute__ ((unused)) = \
209
209
TouchOpRegistrar_##op_type()
210
210
211
- #define USE_OP_DEVICE_KERNEL (op_type, DEVICE_TYPE ) \
212
- STATIC_ASSERT_GLOBAL_NAMESPACE ( \
213
- __use_op_kernel_##op_type##_##DEVICE_TYPE ##__, \
214
- " USE_OP_DEVICE_KERNEL must be in global namespace" ); \
215
- extern int TouchOpKernelRegistrar_##op_type##_##DEVICE_TYPE (); \
216
- static int use_op_kernel_##op_type##_##DEVICE_TYPE ##_ \
217
- __attribute__ ((unused)) = \
218
- TouchOpKernelRegistrar_##op_type##_##DEVICE_TYPE ()
211
+ #define USE_OP_DEVICE_KERNEL (op_type, LIBRARY_TYPE ) \
212
+ STATIC_ASSERT_GLOBAL_NAMESPACE ( \
213
+ __use_op_kernel_##op_type##_##LIBRARY_TYPE ##__, \
214
+ " USE_OP_DEVICE_KERNEL must be in global namespace" ); \
215
+ extern int TouchOpKernelRegistrar_##op_type##_##LIBRARY_TYPE (); \
216
+ static int use_op_kernel_##op_type##_##LIBRARY_TYPE ##_ \
217
+ __attribute__ ((unused)) = \
218
+ TouchOpKernelRegistrar_##op_type##_##LIBRARY_TYPE ()
219
219
220
220
// TODO(fengjiayi): The following macros
221
221
// seems ugly, do we have better method?
0 commit comments