!5262 fix grad all ci bug

Merge pull request !5262 from flywind/fix_grad_all
5 years ago · c0e02bd88b
parent 87643d7a05 44b738e8ac
commit c0e02bd88b
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):