multirow indentity preaggregate dense

apache · Jan 17, 2025 · 44b9e43 · 44b9e43
1 parent cf17891
commit 44b9e43
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/org/apache/sysds/runtime/compress/colgroup/ColGroupDDC.java b/src/main/java/org/apache/sysds/runtime/compress/colgroup/ColGroupDDC.java
@@ -945,12 +945,12 @@ private void leftMMIdentityPreAggregateDenseSingleRow(double[] values, int pos,
 				for(int rc = cl; rc < cu; rc++, pos++) {
 					final int idx = _data.getIndex(rc);
 					if(idx != nVal)
-						values2[_colIndexes.get(idx)] += values[pos];
+						values2[pos2 + _colIndexes.get(idx)] += values[pos];
 				}
 			}
 			else {
 				for(int rc = cl; rc < cu; rc++, pos++)
-					values2[_colIndexes.get(_data.getIndex(rc))] += values[pos];
+					values2[pos2 + _colIndexes.get(_data.getIndex(rc))] += values[pos];
 			}
 		}
 	}
@@ -959,7 +959,7 @@ private void leftMMIdentityPreAggregateDenseSingleRowRangeIndex(double[] values,
 		int cl, int cu) {
 		IdentityDictionary a = (IdentityDictionary) _dict;
 
-		final int firstCol = _colIndexes.get(0);
+		final int firstCol = pos2 + _colIndexes.get(0);
 		pos += cl; // left side matrix position offset.
 		if(a.withEmpty()) {
 			final int nVal = _dict.getNumberOfValues(_colIndexes.size()) - 1;