diff --git a/imperative/python/megengine/autodiff/grad_manager.py b/imperative/python/megengine/autodiff/grad_manager.py index d2d73f3d..19b48eb1 100644 --- a/imperative/python/megengine/autodiff/grad_manager.py +++ b/imperative/python/megengine/autodiff/grad_manager.py @@ -262,7 +262,7 @@ class GradManager: if dy is None: dys = [ones_like(y) for y in ys] elif isinstance(dy, (tuple, list)): - dys = ys + dys = dy else: dys = [dy] try: