[MINOR] Parallel Binary exam sparsity

apache · Feb 3, 2025 · 4e6b5d3 · 4e6b5d3
1 parent ebe758d
commit 4e6b5d3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main/java/org/apache/sysds/runtime/instructions/cp/BinaryMatrixMatrixCPInstruction.java b/src/main/java/org/apache/sysds/runtime/instructions/cp/BinaryMatrixMatrixCPInstruction.java
@@ -93,8 +93,10 @@ public void processInstruction(ExecutionContext ec) {
 			// Release the memory occupied by input matrices
 			ec.releaseMatrixInput(input1.getName(), input2.getName());
 			// Ensure right dense/sparse output representation (guarded by released input memory)
-			if(checkGuardedRepresentationChange(inBlock1, inBlock2, retBlock))
-				retBlock.examSparsity();
+			if(checkGuardedRepresentationChange(inBlock1, inBlock2, retBlock)){
+				int k = (_optr instanceof BinaryOperator) ? ((BinaryOperator) _optr).getNumThreads() : 1; 
+				retBlock.examSparsity(k);
+			}
 		}
 
 		// Attach result matrix with MatrixObject associated with output_name