Skip to content

Commit dc954d2

Browse files
committed
Always use accum for mask.
1 parent bba8dfe commit dc954d2

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

cfpq_model/label_decomposed_graph.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff 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],

0 commit comments

Comments
 (0)