diff --git a/python/paddle/fluid/tests/unittests/prim/composite_ops/test_composite_relu_custom_vjp.py b/python/paddle/fluid/tests/unittests/prim/composite_ops/test_composite_relu_custom_vjp.py index 8113ddee89a29c5571deb0077d8513022fe278a1..65ab814486a3f83b640d9caecb575930a33be5e9 100644 --- a/python/paddle/fluid/tests/unittests/prim/composite_ops/test_composite_relu_custom_vjp.py +++ b/python/paddle/fluid/tests/unittests/prim/composite_ops/test_composite_relu_custom_vjp.py @@ -65,8 +65,8 @@ def expect_grad(inputs): return gradients -class TestCompositeSoftmaxPrimBackward(unittest.TestCase): - "test composite softmax and prim backward" +class TestCompositeReluPrimBackward(unittest.TestCase): + "test composite relu and prim backward" def setUp(self): core._set_prim_backward_enabled(True)