revise bug for concat with quantize

kendryte · Jan 18, 2025 · d71aacc · d71aacc
1 parent b530435
commit d71aacc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/Nncase.Passes/Rules/Neutral/CombineQuantize.cs b/src/Nncase.Passes/Rules/Neutral/CombineQuantize.cs
@@ -48,6 +48,11 @@ public sealed partial class CombineQuantizeConcat : RewriteRule<Pattern>
             // see UnitTestCombineQuantize.TestCombineQuantizeConcatNegative
             foreach (var e in tupleInputs)
             {
+                if (e.CheckedShape.Any(s => s.Value == 0))
+                {
+                    return null;
+                }
+
                 if (userAnalysis[e].Count() > 1)
                 {
                     foreach (var user in userAnalysis[e])