constrain xpu boxing's cost (#1288)

kendryte · Jan 8, 2025 · 64d5095 · 64d5095
1 parent ae1ef36
commit 64d5095
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/modules/Nncase.Modules.CPU/Evaluator/CPU/Boxing.cs b/modules/Nncase.Modules.CPU/Evaluator/CPU/Boxing.cs
@@ -176,6 +176,11 @@ public Cost Visit(ICostEvaluateContext context, Boxing target)
                         {
                             [CostFactorNames.MemoryStore] = (UInt128)((gatherPart - 1) * (float)CostUtility.GetMemoryAccess(DistributedUtility.GetDividedTensorType(a)) / gatherPart),
                         };
+
+                        if (a.Placement.HierarchyKind == HierarchyKind.SMT && (a.NdSBP[1] is SBPPartialSum))
+                        {
+                            cost[CostFactorNames.MemoryStore] *= 8;
+                        }
                     }
 
                     if (scatterPart > 1f)