You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@systemds.apache.org by ba...@apache.org on 2024/08/14 10:07:52 UTC

(systemds) branch main updated: [MINOR] avoid counting non-zeros in sequence datagen

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

baunsgaard 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 bee4232d3f [MINOR] avoid counting non-zeros in sequence datagen
bee4232d3f is described below

commit bee4232d3f65b5dca5cebc81db7fc4b86a61641c
Author: Sebastian Baunsgaard <ba...@apache.org>
AuthorDate: Wed Aug 14 12:07:17 2024 +0200

    [MINOR] avoid counting non-zeros in sequence datagen
---
 .../org/apache/sysds/runtime/matrix/data/LibMatrixDatagen.java     | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/apache/sysds/runtime/matrix/data/LibMatrixDatagen.java b/src/main/java/org/apache/sysds/runtime/matrix/data/LibMatrixDatagen.java
index 18e6b8f0de..c934ca02ad 100644
--- a/src/main/java/org/apache/sysds/runtime/matrix/data/LibMatrixDatagen.java
+++ b/src/main/java/org/apache/sysds/runtime/matrix/data/LibMatrixDatagen.java
@@ -373,7 +373,12 @@ public class LibMatrixDatagen
 			cur += incr;
 		}
 		
-		out.recomputeNonZeros();
+		if((incr > 0 && from > 0) || (incr < 0 && from < 0))
+			out.nonZeros = rows;
+		else if(from == 0 && incr != 0)
+			out.nonZeros = rows - 1;
+		else
+			out.recomputeNonZeros();
 	}
 
 	/**