From b9c863723870ecec231ca0ff0b84e7beab8fe5ce Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Thu, 28 Sep 2017 11:09:42 -0700
Subject: [PATCH] Fix compile

---
 paddle/operators/concat_op.h | 2 +-
 paddle/operators/reduce_op.h | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/paddle/operators/concat_op.h b/paddle/operators/concat_op.h
index bff453971a..c113f19fb5 100644
--- a/paddle/operators/concat_op.h
+++ b/paddle/operators/concat_op.h
@@ -44,7 +44,7 @@ class ConcatKernel : public framework::OpKernel<T> {
 };
 
 template <typename Place, typename T>
-class ConcatGradKernel : public framework::OpKernel {
+class ConcatGradKernel : public framework::OpKernel<T> {
  public:
   void Compute(const framework::ExecutionContext& ctx) const {
     auto* in = ctx.Input<framework::Tensor>(framework::GradVarName("Out"));
diff --git a/paddle/operators/reduce_op.h b/paddle/operators/reduce_op.h
index 2fbf94e34f..ba3f3db81d 100644
--- a/paddle/operators/reduce_op.h
+++ b/paddle/operators/reduce_op.h
@@ -87,7 +87,7 @@ struct MaxOrMinGradFunctor {
 };
 
 template <typename Place, typename T, typename Functor>
-class ReduceKernel : public framework::OpKernel {
+class ReduceKernel : public framework::OpKernel<T> {
  public:
   void Compute(const framework::ExecutionContext& context) const override {
     int rank = context.Input<Tensor>("X")->dims().size();
@@ -141,7 +141,7 @@ class ReduceKernel : public framework::OpKernel {
 };
 
 template <typename Place, typename T, typename Functor>
-class ReduceGradKernel : public framework::OpKernel {
+class ReduceGradKernel : public framework::OpKernel<T> {
  public:
   void Compute(const framework::ExecutionContext& context) const override {
     int rank = context.Input<Tensor>("X")->dims().size();
-- 
GitLab