Skip to content

Commit 26fd1ff

Browse files
committed
Add memory constraints correctly with MiniMalloc
1 parent 87f0135 commit 26fd1ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Deeploy/TilingExtension/MemoryScheduler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -553,7 +553,7 @@ def constraintTileBuffersWithOverlappingLifetime(self, tilerModel: TilerModel, c
553553
if memoryLevel.name == infoDict['memoryLevel']:
554554
sumExpr += infoDict['sizeVar'] * infoDict['typeWidthFactor'] * infoDict['multiBufferCoeff']
555555
if sumExpr != 0:
556-
tilerModel.addConstraint(sumExpr + constantTensorOffset <= memoryLevel.size)
556+
tilerModel.addConstraint(sumExpr + constantTensorOffset, memoryLevel = memoryLevel)
557557

558558
def getSymbolicCostName(self, patternIdx: int, memoryLevel: str) -> str:
559559
stringSuffix = self._stringSuffix + f"_{memoryLevel}"

0 commit comments

Comments
 (0)