prelu modify

8 years ago · 260026fa67
parent 58b5b08bba
commit 260026fa67
2 changed files with 4 additions and 1 deletions
--- 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>);
--- 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")}