File tree Expand file tree Collapse file tree 1 file changed +8
-8
lines changed
Expand file tree Collapse file tree 1 file changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -262,14 +262,14 @@ def mxm(
262262 if swap_operands :
263263 rhs1 , rhs2 = rhs2 , rhs1
264264 if rhs1 in self .matrices and rhs2 in other .matrices :
265- if swap_operands :
266- mask = (accum .matrices [lhs ].to_mask ()
267- if lhs in accum .matrices and other .matrices [rhs2 ].shape == self .matrices [rhs1 ].shape
268- else None )
269- else :
270- mask = (self .matrices [lhs ].to_mask ()
271- if lhs in self .matrices and other .matrices [rhs2 ].shape == self .matrices [rhs1 ].shape
272- else None )
265+ # if swap_operands:
266+ mask = (accum .matrices [lhs ].to_mask ()
267+ if lhs in accum .matrices and other .matrices [rhs2 ].shape == self .matrices [rhs1 ].shape
268+ else None )
269+ # else:
270+ # mask = (self.matrices[lhs].to_mask()
271+ # if lhs in self.matrices and other.matrices[rhs2].shape == self.matrices[rhs1].shape
272+ # else None)
273273
274274 mxm = self .matrices [rhs1 ].mxm (
275275 other .matrices [rhs2 ],
You can’t perform that action at this time.
0 commit comments