Skip to content

Commit 9e9df9b

Browse files
committed
revert gemmtileconstraint
1 parent 8db9601 commit 9e9df9b

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

Deeploy/Targets/PULPOpen/TileConstraints/GEMMTileConstraint.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,6 +193,7 @@ class FloatGEMMTileConstraint(TileConstraint):
193193
@staticmethod
194194
def addGeometricalConstraint(tilerModel: TilerModel, parseDict: Dict, ctxt: NetworkContext) -> TilerModel:
195195

196+
# Get to-be-tiled tensor's buffers
196197
bufferA = ctxt.lookup(name = parseDict['A'])
197198
bufferB = ctxt.lookup(name = parseDict['B'])
198199
outputBuffer = ctxt.lookup(name = parseDict['data_out'])
@@ -223,9 +224,11 @@ def addGeometricalConstraint(tilerModel: TilerModel, parseDict: Dict, ctxt: Netw
223224
outputFirstDimVar = tilerModel.getTensorDimVar(tensorName = outputBuffer.name, dimIdx = dimOffsetOut)
224225
outputSecondDimVar = tilerModel.getTensorDimVar(tensorName = outputBuffer.name, dimIdx = dimOffsetOut + 1)
225226

227+
# Map output dims to inputs dims
226228
tilerModel.addConstraint(outputFirstDimVar == AFirstDimVar)
227229
tilerModel.addConstraint(outputSecondDimVar == BSecondDimVar)
228230

231+
# Add GEMM Geometrical constraints
229232
tilerModel.addConstraint(ASecondDimVar == BFirstDimVar)
230233

231234
# Add bias constraints only if bias is present

0 commit comments

Comments
 (0)