speed up join 10x

breandan · Apr 21, 2023 · f9a9948 · f9a9948
1 parent 8d43532
commit f9a9948
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -25,7 +25,7 @@ dependencies {
 //  implementation("ai.djl.tensorflow:tensorflow-native-cu101:2.3.1")
 //  implementation("ai.djl:examples:0.6.0")
 
-  val djlVersion = "0.22.0"
+  val djlVersion = "0.22.1"
   implementation("ai.djl:api:$djlVersion")
   implementation("ai.djl.mxnet:mxnet-engine:$djlVersion")
   implementation("ai.djl.mxnet:mxnet-native-cu102mkl:1.9.1")
@@ -54,7 +54,7 @@ dependencies {
 
   // Source code transformation
 //  implementation("com.github.h0tk3y.betterParse:better-parse:0.4.2")
-  val openrwVersion = "7.39.1"
+  val openrwVersion = "7.40.0"
   implementation("org.openrewrite:rewrite-java:$openrwVersion")
   runtimeOnly("org.openrewrite:rewrite-java-11:$openrwVersion")
 
@@ -148,7 +148,7 @@ dependencies {
   implementation("com.github.ben-manes.caffeine:caffeine:3.1.6")
 
   // Source Code Transformations
-  implementation("fr.inria.gforge.spoon:spoon-core:10.4.0-beta-1")
+  implementation("fr.inria.gforge.spoon:spoon-core:10.4.0-beta-2")
 
 //  implementation("com.theokanning.openai-gpt3-java:api:0.12.0")
   implementation("com.aallam.openai:openai-client:3.2.1")

diff --git a/src/main/kotlin/edu/mcgill/cstk/experiments/repair/PythonStatementRepair.kt b/src/main/kotlin/edu/mcgill/cstk/experiments/repair/PythonStatementRepair.kt
@@ -61,8 +61,8 @@ fun organicErrorCorrection() {
 private fun optRepair(clock: TimeSource.Monotonic.ValueTimeMark): CFG.(List<Σᐩ>) -> Sequence<Σᐩ> =
   { a: List<Σᐩ> ->
     val timeIsLeft = { clock.elapsedNow().inWholeMilliseconds < TIMEOUT_MS }
-    if(a.count { it.isHoleTokenIn(this) } < 4)
-      a.solve(this, takeMoreWhile = timeIsLeft )
+    if(a.isSetValiantOptimalFor(this))
+      a.solve(this, takeMoreWhile = timeIsLeft)
     else asCJL.synthesize(a, takeMoreWhile = timeIsLeft)
   }