fix grad all ci bug

5 years ago · 44b738e8ac
parent 13e9f5b86e
commit 44b738e8ac
1 changed files with 1 additions and 1 deletions
--- a/tests/ut/python/pipeline/parse/test_parse.py
+++ b/tests/ut/python/pipeline/parse/test_parse.py
@ -277,7 +277,7 @@ class AssignWhenInsertGrad(nn.Cell):
        out = self.getG(out)
        return out

-grad_all = C.GradOperation('get_all', get_all=True)
+grad_all = C.GradOperation(get_all=True)

 class GradNet(nn.Cell):
    def __init__(self, net):