From 260026fa678177e7f21390fd560422de5e1b046e Mon Sep 17 00:00:00 2001
From: zchen0211 <chenzhuoyuan07@gmail.com>
Date: Thu, 14 Sep 2017 17:23:47 -0700
Subject: [PATCH] prelu modify

---
 paddle/operators/prelu_op.cu                      | 3 +++
 python/paddle/v2/framework/tests/test_prelu_op.py | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/paddle/operators/prelu_op.cu b/paddle/operators/prelu_op.cu
index 54a9089bd..314dcba37 100644
--- a/paddle/operators/prelu_op.cu
+++ b/paddle/operators/prelu_op.cu
@@ -16,3 +16,6 @@
 
 REGISTER_OP_GPU_KERNEL(
     prelu, paddle::operators::PreluKernel<paddle::platform::GPUPlace, float>);
+REGISTER_OP_GPU_KERNEL(
+    prelu_grad,
+    paddle::operators::PreluGradKernel<paddle::platform::GPUPlace, float>);
diff --git a/python/paddle/v2/framework/tests/test_prelu_op.py b/python/paddle/v2/framework/tests/test_prelu_op.py
index 8b3916696..c207940d1 100644
--- a/python/paddle/v2/framework/tests/test_prelu_op.py
+++ b/python/paddle/v2/framework/tests/test_prelu_op.py
@@ -3,7 +3,7 @@ import numpy as np
 from op_test import OpTest
 
 
-class ScaleTest(OpTest):
+class PreluTest(OpTest):
     def setUp(self):
         self.op_type = "prelu"
         self.inputs = {'X': np.random.random((10, 10)).astype("float32")}
-- 
GitLab