From 44b738e8ac359c5ee252ae88ebf6eecc499a87ee Mon Sep 17 00:00:00 2001 From: kpy Date: Wed, 26 Aug 2020 18:28:35 +0800 Subject: [PATCH] fix grad all ci bug --- tests/ut/python/pipeline/parse/test_parse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/ut/python/pipeline/parse/test_parse.py b/tests/ut/python/pipeline/parse/test_parse.py index db0fa42162..3a8e182b84 100644 --- 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):