Bugfix fix

Jaybit0 · Jan 29, 2025 · ef49168 · ef49168
1 parent 8fca8e6
commit ef49168
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/main/java/org/apache/sysds/hops/rewriter/RewriterRuleCollection.java b/src/main/java/org/apache/sysds/hops/rewriter/RewriterRuleCollection.java
@@ -1675,9 +1675,7 @@ public static void canonicalExpandAfterFlattening(final List<RewriterRule> rules
 								.as(UUID.randomUUID().toString())
 								.withInstruction("sum")
 								.withOps(newIdxExpr);
-						System.out.println("Copying index list: " + newIdxExpr.toParsableString(ctx));
 						RewriterUtils.copyIndexList(newIdxExpr);
-						System.out.println("After copy: " + newIdxExpr.toParsableString(ctx));
 						newIdxExpr.refreshReturnType(ctx);
 						newSum.consolidate(ctx);
 						newArgList.getOperands().add(newSum);

diff --git a/src/main/java/org/apache/sysds/hops/rewriter/utils/RewriterUtils.java b/src/main/java/org/apache/sysds/hops/rewriter/utils/RewriterUtils.java
@@ -869,7 +869,8 @@ public static void copyIndexList(RewriterStatement idxExprRoot) {
 			return null;
 		});
 
-		idxExprRoot.getOperands().set(1, out);
+		if (out != null)
+			idxExprRoot.getOperands().set(1, out);
 	}
 
 	public static void retargetIndexExpressions(RewriterStatement rootExpr, UUID oldIdxId, RewriterStatement newStatement) {