File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -350,6 +350,9 @@ def parseNodeCtxt(self,
350350class 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:
358361class 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 ])
You can’t perform that action at this time.
0 commit comments