Skip to content

Commit

Permalink
Follow comments, change KT to KERNEL_TYPE
Browse files Browse the repository at this point in the history
  • Loading branch information
reyoung committed Sep 28, 2017
1 parent 87da154 commit d53b38e
Showing 1 changed file with 8 additions and 7 deletions.
15 changes: 8 additions & 7 deletions paddle/framework/op_registry.h
Original file line number Diff line number Diff line change
Expand Up @@ -103,18 +103,19 @@ class OpRegistrar : public Registrar {
template <typename PlaceType, bool at_end, size_t I, typename... KernelType>
struct OpKernelRegistrarFunctor;

template <typename PlaceType, size_t I, typename... KernelType>
struct OpKernelRegistrarFunctor<PlaceType, false, I, KernelType...> {
using KT = typename std::tuple_element<I, std::tuple<KernelType...>>::type;
template <typename PlaceType, size_t I, typename... KernelTypes>
struct OpKernelRegistrarFunctor<PlaceType, false, I, KernelTypes...> {
using KERNEL_TYPE =
typename std::tuple_element<I, std::tuple<KernelTypes...>>::type;

void operator()(const char* op_type) const {
using T = typename KT::ELEMENT_TYPE;
using T = typename KERNEL_TYPE::ELEMENT_TYPE;
OperatorWithKernel::OpKernelKey key(ToDataType(std::type_index(typeid(T))),
PlaceType());
OperatorWithKernel::AllOpKernels()[op_type][key].reset(new KT);
OperatorWithKernel::AllOpKernels()[op_type][key].reset(new KERNEL_TYPE);

constexpr auto size = std::tuple_size<std::tuple<KernelType...>>::value;
OpKernelRegistrarFunctor<PlaceType, I + 1 == size, I + 1, KernelType...>
constexpr auto size = std::tuple_size<std::tuple<KernelTypes...>>::value;
OpKernelRegistrarFunctor<PlaceType, I + 1 == size, I + 1, KernelTypes...>
func;
func(op_type);
}
Expand Down

0 comments on commit d53b38e

Please sign in to comment.