diff --git a/paddle/operators/prelu_op.cu b/paddle/operators/prelu_op.cu index 54a9089bdb66c47cbe21d87d2f6813b7aec4a299..314dcba375c89c7a752893c16f62a09734d609fb 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); +REGISTER_OP_GPU_KERNEL( + prelu_grad, + paddle::operators::PreluGradKernel); diff --git a/python/paddle/v2/framework/tests/test_prelu_op.py b/python/paddle/v2/framework/tests/test_prelu_op.py index 8b3916696a69c958d5609ecdf961a25c28352184..c207940d1f68ceedc9546d82a178efb39fa5d3ef 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")}