Avoid recursion error in profiling

xadupre · xadupre · commit b94eb3f440ef · 2024-12-20T13:55:38.000+01:00
diff --git a/onnx_array_api/profiling.py b/onnx_array_api/profiling.py
@@ -73,8 +73,8 @@ def _get_root(node, stor=None):
                 stor.append(node)
             if not node.called_by:
                 return node
-            if len(node.called_by) == 1:
-                return _get_root(node.called_by[0], stor=stor)
+            if len(node.called_by) == 0:
+                return None
             res = None
             for ct in node.called_by:
                 k = id(node), id(ct)
