fix auot-dist with nop output

kendryte · Sep 26, 2024 · c4e916b · c4e916b
1 parent 2e28739
commit c4e916b
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/modules/Nncase.Modules.CPU/Passes/Distributed/AutoDistributed.cs b/modules/Nncase.Modules.CPU/Passes/Distributed/AutoDistributed.cs
@@ -185,11 +185,19 @@ void Dfs(IEquality equality)
                 var outputs = typeEquivalents.Select(g => InstertTerminator(g.Value[0]))
                 .Select(e => new EqualityNode(e))
                 .OfType<IEquality>().ToList();
-                _equalMemo.Add(function.Body, new EqualityClass(false, outputs));
 
-                using (new ExprPinner(outputs.Select(e => ((EqualityNode)e).Expr).ToArray()))
+                if (outputs.Any())
                 {
-                    BranchCut();
+                    _equalMemo.Add(function.Body, new EqualityClass(false, outputs));
+
+                    using (new ExprPinner(outputs.Select(e => ((EqualityNode)e).Expr).ToArray()))
+                    {
+                        BranchCut();
+                    }
+                }
+                else
+                {
+                    return input;
                 }
             }