Skip to content

Commit d53b38e

Browse files
committed
Follow comments, change KT to KERNEL_TYPE
1 parent 87da154 commit d53b38e

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

paddle/framework/op_registry.h

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -103,18 +103,19 @@ class OpRegistrar : public Registrar {
103103
template <typename PlaceType, bool at_end, size_t I, typename... KernelType>
104104
struct OpKernelRegistrarFunctor;
105105

106-
template <typename PlaceType, size_t I, typename... KernelType>
107-
struct OpKernelRegistrarFunctor<PlaceType, false, I, KernelType...> {
108-
using KT = typename std::tuple_element<I, std::tuple<KernelType...>>::type;
106+
template <typename PlaceType, size_t I, typename... KernelTypes>
107+
struct OpKernelRegistrarFunctor<PlaceType, false, I, KernelTypes...> {
108+
using KERNEL_TYPE =
109+
typename std::tuple_element<I, std::tuple<KernelTypes...>>::type;
109110

110111
void operator()(const char* op_type) const {
111-
using T = typename KT::ELEMENT_TYPE;
112+
using T = typename KERNEL_TYPE::ELEMENT_TYPE;
112113
OperatorWithKernel::OpKernelKey key(ToDataType(std::type_index(typeid(T))),
113114
PlaceType());
114-
OperatorWithKernel::AllOpKernels()[op_type][key].reset(new KT);
115+
OperatorWithKernel::AllOpKernels()[op_type][key].reset(new KERNEL_TYPE);
115116

116-
constexpr auto size = std::tuple_size<std::tuple<KernelType...>>::value;
117-
OpKernelRegistrarFunctor<PlaceType, I + 1 == size, I + 1, KernelType...>
117+
constexpr auto size = std::tuple_size<std::tuple<KernelTypes...>>::value;
118+
OpKernelRegistrarFunctor<PlaceType, I + 1 == size, I + 1, KernelTypes...>
118119
func;
119120
func(op_type);
120121
}

0 commit comments

Comments
 (0)