Skip to content

Commit a8c500b

Browse files
committed
Fix missing super parseNode call
1 parent da18e13 commit a8c500b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

Deeploy/Targets/PULPOpen/Parsers.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -350,6 +350,9 @@ def parseNodeCtxt(self,
350350
class PULPMatrixVecParser(PULPGEMMParser):
351351

352352
def parseNode(self, node: gs.Node) -> bool:
353+
if not super().parseNode(node):
354+
return False
355+
353356
M = node.inputs[0].shape[-1 if node.attrs["transA"] else -2]
354357
batch = math.prod(node.inputs[0].shape[:-2])
355358
return super().parseNode(node) and M == 1 and batch >= 8
@@ -358,6 +361,9 @@ def parseNode(self, node: gs.Node) -> bool:
358361
class PULPTallGEMMParser(PULPGEMMParser):
359362

360363
def parseNode(self, node: gs.Node) -> bool:
364+
if not super().parseNode(node):
365+
return False
366+
361367
M = node.inputs[0].shape[-1 if node.attrs["transA"] else -2]
362368
N = node.inputs[1].shape[-2 if node.attrs["transB"] else -1]
363369
batch = math.prod(node.inputs[0].shape[:-2])

0 commit comments

Comments
 (0)