[MINOR] Fix consistent nnz maintenance in codegen col aggregates

Frxms · mboehm7 · commit 4949e96ba9ae · 2025-02-01T10:35:47.000+01:00
Closes #2205.
diff --git a/src/main/java/org/apache/sysds/runtime/codegen/SpoofCellwise.java b/src/main/java/org/apache/sysds/runtime/codegen/SpoofCellwise.java
@@ -267,6 +267,8 @@ public MatrixBlock execute(ArrayList<MatrixBlock> inputs, ArrayList<ScalarObject
 				lnnz = executeDense(a.getDenseBlock(), b, scalars, out, m, n, sparseSafe, 0, m, rix);
 			else
 				lnnz = executeSparse(a.getSparseBlock(), b, scalars, out, m, n, sparseSafe, 0, m, rix);
+			if(_type == CellType.COL_AGG)
+				lnnz = out.recomputeNonZeros();
 		}
 		else  //MULTI-THREADED
 		{

Original file line number	Diff line number	Diff line change
`@@ -267,6 +267,8 @@ public MatrixBlock execute(ArrayList<MatrixBlock> inputs, ArrayList<ScalarObject`
`267`	`267`	`lnnz = executeDense(a.getDenseBlock(), b, scalars, out, m, n, sparseSafe, 0, m, rix);`
`268`	`268`	`else`
`269`	`269`	`lnnz = executeSparse(a.getSparseBlock(), b, scalars, out, m, n, sparseSafe, 0, m, rix);`
	`270`	`+ if(_type == CellType.COL_AGG)`
	`271`	`+ lnnz = out.recomputeNonZeros();`
`270`	`272`	`}`
`271`	`273`	`else //MULTI-THREADED`
`272`	`274`	`{`