mindstudio-probe 8.3.2__py3-none-any.whl → 8.3.3__py3-none-any.whl

{mindstudio_probe-8.3.2.dist-info → mindstudio_probe-8.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mindstudio-probe
-Version: 8.3.2
+Version: 8.3.3
 Summary: Ascend Probe Utils
 Home-page: https://gitcode.com/Ascend/mstt/tree/master/debug/accuracy_tools/msprobe
 Author: Ascend Team

{mindstudio_probe-8.3.2.dist-info → mindstudio_probe-8.3.3.dist-info}/RECORD

@@ -319,7 +319,7 @@ msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py,sha256=ZzVF13owq
 msprobe/mindspore/overflow_check/overflow_check_tool_factory.py,sha256=6JqOsKCaNZbUi_pXCv6uGqHQX4yhhauqoPziyI97FUw,1909
 msprobe/nan_analyze/__init__.py,sha256=S8S4SIEKDrF4pk8alDPapKUdlTHzok2jFc5kqk-JIL0,617
 msprobe/nan_analyze/analyzer.py,sha256=EEMwQWm33JgXgvLudoMRkIYDQpQaqViFpl6BFMIrOtM,11667
-msprobe/nan_analyze/graph.py,sha256=uSEYUC4I1KhGPUyi3E1BblNNpxUY3fjpf_sNnaaY7w4,8085
+msprobe/nan_analyze/graph.py,sha256=PwuJL62Sgg7ANZKpCPMwPxx41cxvUiXZMDkLgkTUUI8,8164
 msprobe/nan_analyze/utils.py,sha256=jwpnHcfluzBlytAcUuyAotj0Hw3zmi7e8ncf7rME2JU,7666
 msprobe/pytorch/__init__.py,sha256=qIvhnAk61oSpvPU0QI0YAC4zyLKYyOcfNzdXJxN7Klo,1035
 msprobe/pytorch/function_factory.py,sha256=Fi4w0zfO64Sd2IU9z45mBbDgS8k_CQEiZ9vRpC_TkVk,4031
@@ -483,9 +483,9 @@ msprobe/visualization/graph/distributed_analyzer.py,sha256=IVPaA6Z3uZxuOob7fibgc
 msprobe/visualization/graph/graph.py,sha256=SFf_x4a3e5ejUe6PntPeEqnGCPaR6gUWpak6w_hyIBE,8776
 msprobe/visualization/graph/node_colors.py,sha256=7LpurTuE3edKGilwLVsXD7Ue4bMT7Maz63Udq_6CADM,4504
 msprobe/visualization/graph/node_op.py,sha256=qkbw3ZJkKGQH071C1CLIXi4kavvsLxBTzt0KXncz6fY,1368
-mindstudio_probe-8.3.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mindstudio_probe-8.3.2.dist-info/METADATA,sha256=wkdXr0lpQ-KJB2H3HR5Y7gEpXPIEQc0u6RailLM3cbA,1437
-mindstudio_probe-8.3.2.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-mindstudio_probe-8.3.2.dist-info/entry_points.txt,sha256=4ob3a9L018EBZFdlfgMW1lbgeIOhc4F-HCR8gBksaCQ,49
-mindstudio_probe-8.3.2.dist-info/top_level.txt,sha256=LxFEFqelENSyWmRtocCiEUF04IE8aZvwTl7ADB598Tk,8
-mindstudio_probe-8.3.2.dist-info/RECORD,,
+mindstudio_probe-8.3.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mindstudio_probe-8.3.3.dist-info/METADATA,sha256=zb5GEQTdL6OI9SvkeSJDwFk9h_gNNcC7GtrA4pOKseE,1437
+mindstudio_probe-8.3.3.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+mindstudio_probe-8.3.3.dist-info/entry_points.txt,sha256=4ob3a9L018EBZFdlfgMW1lbgeIOhc4F-HCR8gBksaCQ,49
+mindstudio_probe-8.3.3.dist-info/top_level.txt,sha256=LxFEFqelENSyWmRtocCiEUF04IE8aZvwTl7ADB598Tk,8
+mindstudio_probe-8.3.3.dist-info/RECORD,,

msprobe/nan_analyze/graph.py

@@ -46,6 +46,8 @@ class DataNode:
         seen = set(op_name)
         while True:
             op_name = construct_info.get(op_name)
+            if isinstance(op_name, list):
+                op_name = op_name[0]
             if not op_name or op_name in seen:
                 return construct
             construct.insert(0, op_name)