(systemds) branch main updated: [MINOR] Fix consistent nnz maintenance in codegen col aggregates

[mboehm7]

This is an automated email from the ASF dual-hosted git repository.

mboehm7 pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/systemds.git


The following commit(s) were added to refs/heads/main by this push:
     new 4949e96ba9 [MINOR] Fix consistent nnz maintenance in codegen col 
aggregates
4949e96ba9 is described below

commit 4949e96ba9aeb3f14b353e67ec8e1c7471d140f2
Author: Frxms <tomoki.men...@gmail.com>
AuthorDate: Sat Feb 1 10:35:46 2025 +0100

    [MINOR] Fix consistent nnz maintenance in codegen col aggregates
    
    Closes #2205.
---
 src/main/java/org/apache/sysds/runtime/codegen/SpoofCellwise.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/main/java/org/apache/sysds/runtime/codegen/SpoofCellwise.java 
b/src/main/java/org/apache/sysds/runtime/codegen/SpoofCellwise.java
index 73967154c6..6c3986bd93 100644
--- 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 abstract class SpoofCellwise extends SpoofOperator {
                                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
                {