From 01354337a948d6be8c0a4fbf20a4fc9ea6ad9acc Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 3 Jun 2021 20:05:22 +0800
Subject: [PATCH] fix(mge/autodiff): fix incorrect handling of tuple dy

GitOrigin-RevId: beca8e37117d1b1f24d463f39903a1a74a62d167
---
 imperative/python/megengine/autodiff/grad_manager.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/python/megengine/autodiff/grad_manager.py b/imperative/python/megengine/autodiff/grad_manager.py
index d2d73f3dd..19b48eb15 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:
-- 
GitLab