Skip to content

Commit 6f71f89

Browse files
authored
change DEVICE_TYPE in op_registry to LIBRARY_TYPE (#7588)
1 parent 144854d commit 6f71f89

File tree

1 file changed

+18
-18
lines changed

1 file changed

+18
-18
lines changed

paddle/framework/op_registry.h

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -177,16 +177,16 @@ class OpKernelRegistrar : public Registrar {
177177
/**
178178
* Macro to register OperatorKernel.
179179
*/
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; \
190190
}
191191

192192
#define REGISTER_OP_CUDA_KERNEL(op_type, ...) \
@@ -208,14 +208,14 @@ class OpKernelRegistrar : public Registrar {
208208
static int use_op_itself_##op_type##_ __attribute__((unused)) = \
209209
TouchOpRegistrar_##op_type()
210210

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()
219219

220220
// TODO(fengjiayi): The following macros
221221
// seems ugly, do we have better method?

0 commit comments

Comments
 (0)