Skip to content

Commit bbd79d2

Browse files
committed
[MINOR] Fix incremental SliceLine pruning configurations
This patch fixes the disabled approximate pruning strategy, as well as moves some of the tests (by correct alphabetical order) to builtin-part1 which also reduces the chances of timeouts.
1 parent 665abc1 commit bbd79d2

File tree

4 files changed

+13
-13
lines changed

4 files changed

+13
-13
lines changed

scripts/builtin/incSliceLine.dml

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,9 @@
5252
# prevTK previous top-k slices (for incremental updates)
5353
# prevTKC previous top-k scores (for incremental updates)
5454
# encodeLat flag for encoding output lattice for less memory consumption
55-
# pruningStrat pruning strategy: 0 all pruning, 1 only score pruning, 2 only size pruning,
56-
# 3 only max score pruning, 4 only approx pruning, 5 no pruning
55+
# pruningStrat pruning strategy: 0 all pruning, 1 all exact pruning,
56+
# 2 only score pruning, 3 only max score pruning,
57+
# 4 only size pruning, 5 no pruning
5758
# ---------------------------------------------------------------------------------------
5859
#
5960
# OUTPUT:
@@ -82,7 +83,7 @@ m_incSliceLine = function(
8283
list[unknown] prevLattice = list(), list[unknown] metaPrevLattice = list(),
8384
list[unknown] prevStats = list(), Matrix[Double] prevTK = matrix(0,0,0),
8485
Matrix[Double] prevTKC = matrix(0,0,0), Boolean encodeLat = TRUE,
85-
Int pruningStrat = 0)
86+
Int pruningStrat = 1)
8687
return(
8788
Matrix[Double] TK, Matrix[Double] TKC, Matrix[Double] D,
8889
list[unknown] L, list[unknown] metaLattice,
@@ -101,11 +102,10 @@ m_incSliceLine = function(
101102
+ " -- see documentation for more details.");
102103
}
103104

104-
enableIncScorePruning = ( pruningStrat <= 1);
105-
enableIncSizePruning = ((pruningStrat == 0) | (pruningStrat == 2));
106-
enableIncMaxScorePruning = ((pruningStrat == 0) | (pruningStrat == 3));
107-
enableIncApproxPruning = ((pruningStrat == 0) | (pruningStrat == 4));
108-
enableIncApproxPruning = FALSE;
105+
enableIncScorePruning = (pruningStrat <= 2);
106+
enableIncSizePruning = (pruningStrat <= 1 | pruningStrat == 4);
107+
enableIncMaxScorePruning = (pruningStrat <= 1 | pruningStrat == 3);
108+
enableIncApproxPruning = (pruningStrat == 0);
109109

110110
t1 = time();
111111

src/test/java/org/apache/sysds/test/functions/builtin/part2/BuiltinDifferenceStatistics.java renamed to src/test/java/org/apache/sysds/test/functions/builtin/part1/BuiltinDifferenceStatistics.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
* under the License.
1818
*/
1919

20-
package org.apache.sysds.test.functions.builtin.part2;
20+
package org.apache.sysds.test.functions.builtin.part1;
2121

2222
import static org.junit.Assert.assertEquals;
2323
import static org.junit.Assert.assertTrue;

src/test/java/org/apache/sysds/test/functions/builtin/part2/BuiltinIncSliceLineTest.java renamed to src/test/java/org/apache/sysds/test/functions/builtin/part1/BuiltinIncSliceLineTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
* under the License.
1818
*/
1919

20-
package org.apache.sysds.test.functions.builtin.part2;
20+
package org.apache.sysds.test.functions.builtin.part1;
2121

2222
import org.junit.Assert;
2323
import org.junit.Test;

src/test/scripts/functions/builtin/incSliceLineFull.dml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,13 +30,13 @@ disableIncScorePruning = $13;
3030
disableIncSizePruning = $14;
3131

3232
if(disableIncScorePruning & disableIncSizePruning){
33-
pruningStrat = 3;
33+
pruningStrat = 5;
3434
} else if (disableIncSizePruning){
3535
pruningStrat = 2;
3636
} else if (disableIncScorePruning){
37-
pruningStrat = 1;
37+
pruningStrat = 4;
3838
} else {
39-
pruningStrat = 0;
39+
pruningStrat = 1;
4040
}
4141

4242

0 commit comments

Comments
 (0)