From 7dee33f93da0675f09784d701ad12c756e3ff917 Mon Sep 17 00:00:00 2001
From: fuchao <fuchao32@huawei.com>
Date: Fri, 20 Jun 2025 12:46:45 +0800
Subject: [PATCH] test

---
 .../mindspore/dump/cell_dump_process.py       | 75 ++++++++++++++-----
 1 file changed, 56 insertions(+), 19 deletions(-)

diff --git a/debug/accuracy_tools/msprobe/mindspore/dump/cell_dump_process.py b/debug/accuracy_tools/msprobe/mindspore/dump/cell_dump_process.py
index c1c25ac938b..01851d6d301 100644
--- a/debug/accuracy_tools/msprobe/mindspore/dump/cell_dump_process.py
+++ b/debug/accuracy_tools/msprobe/mindspore/dump/cell_dump_process.py
@@ -140,25 +140,62 @@ def cell_construct_wrapper(func, self):
         forward_or_all = self.data_mode in ["forward", "all"]
         # The inputs of the cell.
         for index, item in enumerate(args):
-            if backward_or_all and ops.is_tensor(item):
-                if need_tensordump_in(self, 'input_dump_mode', index):
-                    item = gd(gen_file_path(self.dump_path, self.cell_prefix, KEY_BACKWARD, KEY_OUTPUT, index),
-                              item, "out")
-                else:
-                    item = gd(gen_file_path(self.dump_path, self.cell_prefix, KEY_BACKWARD, KEY_OUTPUT, index),
-                              item, "in")
-            if forward_or_all and ops.is_tensor(item):
-                if need_tensordump_in(self, 'input_dump_mode', index):
-                    temp = td_in(
-                        gen_file_path(self.dump_path, self.cell_prefix, KEY_FORWARD, KEY_INPUT, index),
-                        item
-                    )
-                else:
-                    temp = td(
-                        gen_file_path(self.dump_path, self.cell_prefix, KEY_FORWARD, KEY_INPUT, index),
-                        item
-                    )
-                item = ops.depend(item, temp)
+            if isinstance(item, (tuple, list)):
+                print("####FC####item is (tuple, list)")
+                print(self.cell_prefix)
+                print(index)
+                print("####FC#########################tuple")
+                if isinstance(item, list):
+                    print("####FC#########################list")
+            if isinstance(item, (tuple, list)):
+                new_item = []
+                for index_, item_ in enumerate(item):
+                    if isinstance(item_, (tuple, list)):
+                        print("####FC111####item is (tuple, list)")
+                        print(self.cell_prefix)
+                        print(index_)
+                        print("####FC111#########################")
+                    if backward_or_all and ops.is_tensor(item_):
+                        if need_tensordump_in(self, 'input_dump_mode', index_):
+                            item_ = gd(gen_file_path(self.dump_path, self.cell_prefix, KEY_BACKWARD, KEY_OUTPUT, index_),
+                                    item_, "out")
+                        else:
+                            item_ = gd(gen_file_path(self.dump_path, self.cell_prefix, KEY_BACKWARD, KEY_OUTPUT, index_),
+                                    item_, "in")
+                    if forward_or_all and ops.is_tensor(item_):
+                        if need_tensordump_in(self, 'input_dump_mode', index_):
+                            temp = td_in(
+                                gen_file_path(self.dump_path, self.cell_prefix, KEY_FORWARD, KEY_INPUT, index_),
+                                item_
+                            )
+                        else:
+                            temp = td(
+                                gen_file_path(self.dump_path, self.cell_prefix, KEY_FORWARD, KEY_INPUT, index_),
+                                item_
+                            )
+                        item_ = ops.depend(item_, temp)
+                    new_item.append(item_)
+                item = tuple(new_item)
+            else:
+                if backward_or_all and ops.is_tensor(item):
+                    if need_tensordump_in(self, 'input_dump_mode', index):
+                        item = gd(gen_file_path(self.dump_path, self.cell_prefix, KEY_BACKWARD, KEY_OUTPUT, index),
+                                item, "out")
+                    else:
+                        item = gd(gen_file_path(self.dump_path, self.cell_prefix, KEY_BACKWARD, KEY_OUTPUT, index),
+                                item, "in")
+                if forward_or_all and ops.is_tensor(item):
+                    if need_tensordump_in(self, 'input_dump_mode', index):
+                        temp = td_in(
+                            gen_file_path(self.dump_path, self.cell_prefix, KEY_FORWARD, KEY_INPUT, index),
+                            item
+                        )
+                    else:
+                        temp = td(
+                            gen_file_path(self.dump_path, self.cell_prefix, KEY_FORWARD, KEY_INPUT, index),
+                            item
+                        )
+                    item = ops.depend(item, temp)
             new_args.append(item)
 
         out = func(*new_args, **kwargs)
-- 
Gitee