cost estimator allow scalar in table expand

apache · Jan 17, 2025 · a97ceff · a97ceff
1 parent d9ebcf0
commit a97ceff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/org/apache/sysds/resource/cost/CostEstimator.java b/src/main/java/org/apache/sysds/resource/cost/CostEstimator.java
@@ -822,7 +822,7 @@ public double parseSPInst(SPInstruction inst) throws CostEstimationException {
 					SparkCostUtils.getMatMulChainInstTime(mmchaininst, input1, input2, input3, output, driverMetrics, executorMetrics);
 		} else if (inst instanceof CtableSPInstruction) {
 			CtableSPInstruction tableInst = (CtableSPInstruction) inst;
-			VarStats input1 = getStats(tableInst.input1.getName());
+			VarStats input1 = getStatsWithDefaultScalar(tableInst.input1.getName());
 			VarStats input2 = getStatsWithDefaultScalar(tableInst.input2.getName());
 			VarStats input3 = getStatsWithDefaultScalar(tableInst.input3.getName());
 			double loadTime = loadRDDStatsAndEstimateTime(input1) +