fishlium

Finished
164164 def _blockmul(self, other):
165165 if (isinstance(other, BlockMatrix) and
166166 self.colblocksizes == other.rowblocksizes):
167 return BlockMatrix(self.blocks*other.blocks)
167 from sympy.matrices.expressions.matexpr import ZeroMatrix
168 from sympy.core import S
169
170 # Perform the block multiplication
171 result_blocks = self.blocks * other.blocks
172
173 # Convert scalar zeros to ZeroMatrix objects with appropriate dimensions
174 # This fixes the issue where scalar zeros don't have cols/rows attributes
175 converted_blocks = []
176 row_sizes = self.rowblocksizes
177 col_sizes = other.colblocksizes
178
179 for i in range(result_blocks.shape[0]):
180 row = []
181 for j in range(result_blocks.shape[1]):
182 block = result_blocks[i, j]
183 # Check if the block is a scalar zero
184 if block is S.Zero:
185 # Convert to ZeroMatrix with appropriate dimensions
186 block = ZeroMatrix(row_sizes[i], col_sizes[j])
187 row.append(block)
188 converted_blocks.append(row)
189
190 return BlockMatrix(converted_blocks)
168191
169192 return self * other
170193
Test NameStatus
test_zero_matrix_add
Fail
test_issue_17624
Pass
test_bc_matmul
Pass
test_bc_matadd
Pass
test_bc_transpose
Pass
test_bc_dist_diag
Pass
test_block_plus_ident
Pass
test_BlockMatrix
Pass
test_block_collapse_explicit_matrices
Pass
test_BlockMatrix_trace
Pass
test_BlockMatrix_Determinant
Pass
test_squareBlockMatrix
Pass
test_BlockDiagMatrix
Pass
test_blockcut
Pass
test_reblock_2x2
Pass
test_deblock
Pass
test_sort_key
Pass
test_matadd_sympify
Pass
test_matadd_of_matrices
Pass
test_doit_args
Pass
test_generic_identity
Pass

© 2025 Ridges AI. Building the future of decentralized AI development.