[feat] Replace warmup iterations with pre-compilation to ensure JIT compilation from Java to GPU code

mikepapadim · mikepapadim · commit 94b4ec9c188f · 2025-06-09T14:11:20.000+03:00
diff --git a/src/main/java/com/example/tornadovm/TornadoVMMasterPlan.java b/src/main/java/com/example/tornadovm/TornadoVMMasterPlan.java
@@ -66,7 +66,7 @@ public static TornadoVMMasterPlan initializeTornadoVMPlan(State state, Llama mod
         }
 
         // 2. Perform warmup with extra iterations to ensure JIT compilation is complete
-        tornadoVMPlan.executionPlan.withWarmUp(); // Increased warmup iterations
+        tornadoVMPlan.executionPlan.withPreCompilation(); // Force JIT compilation from Java to GPU code
 
         // Record time after warmup
         if (ENABLE_TORNADOVM_INIT_TIME) {

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,7 @@ public static TornadoVMMasterPlan initializeTornadoVMPlan(State state, Llama mod`
`66`	`66`	`}`
`67`	`67`
`68`	`68`	`// 2. Perform warmup with extra iterations to ensure JIT compilation is complete`
`69`		`- tornadoVMPlan.executionPlan.withWarmUp(); // Increased warmup iterations`
	`69`	`+ tornadoVMPlan.executionPlan.withPreCompilation(); // Force JIT compilation from Java to GPU code`
`70`	`70`
`71`	`71`	`// Record time after warmup`
`72`	`72`	`if (ENABLE_TORNADOVM_INIT_TIME) {`