st

apache · Aug 30, 2023 · 182accc · 182accc
1 parent 296137b
commit 182accc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/main/java/org/apache/sysds/runtime/compress/colgroup/scheme/CompressionScheme.java b/src/main/java/org/apache/sysds/runtime/compress/colgroup/scheme/CompressionScheme.java
@@ -185,12 +185,11 @@ public CompressedMatrixBlock updateAndEncode(MatrixBlock mb, int k) {
 		validateInput(mb);
 
 		boolean transposed = false;
-		if(mb.getSparsity() < 0.1){
+		if(mb.getSparsity() < 0.1) {
 			transposed = true;
 			mb = LibMatrixReorg.transpose(mb, k, true);
 		}
 
-
 		final ExecutorService pool = CommonThreadPool.get(k);
 		try {
 			final int nCol = mb.getNumColumns();
@@ -219,9 +218,15 @@ public CompressedMatrixBlock updateAndEncode(MatrixBlock mb) {
 		validateInput(mb);
 
 		List<AColGroup> ret = new ArrayList<>(encodings.length);
+		boolean transposed = false;
+		if(mb.getSparsity() < 0.1) {
+			transposed = true;
+			mb = LibMatrixReorg.transpose(mb, 1 , true);
+		}
 
 		for(int i = 0; i < encodings.length; i++) {
-			Pair<ICLAScheme, AColGroup> p = encodings[i].updateAndEncode(mb);
+			ICLAScheme e = encodings[i];
+			Pair<ICLAScheme, AColGroup> p = transposed ? e.updateAndEncodeT(mb):  e.updateAndEncode(mb);
 			encodings[i] = p.getKey();
 			ret.add(p.getValue());
 		}

diff --git a/src/test/java/org/apache/sysds/test/component/compress/colgroup/scheme/SchemeTestCONST.java b/src/test/java/org/apache/sysds/test/component/compress/colgroup/scheme/SchemeTestCONST.java
@@ -37,7 +37,6 @@
 import org.apache.sysds.runtime.compress.estim.CompressedSizeInfo;
 import org.apache.sysds.runtime.compress.estim.CompressedSizeInfoColGroup;
 import org.apache.sysds.runtime.matrix.data.MatrixBlock;
-import org.apache.sysds.test.TestUtils;
 import org.junit.runner.RunWith;
 import org.junit.runners.Parameterized;
 import org.junit.runners.Parameterized.Parameters;