perf: force sync

LuxDL · Nov 2, 2024 · c35441f · c35441f
1 parent 96e9ce8
commit c35441f
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/perf/resnet/lux_reactant.jl b/perf/resnet/lux_reactant.jl
@@ -43,12 +43,9 @@ function main()
 
         x = rand(Float32, 224, 224, 3, b) |> dev
 
-        model_compiled = @compile model(x, ps_ra, st_ra)
+        model_compiled = Reactant.compile(model, (x, ps_ra, st_ra); block_until_ready=true)
 
-        timings[b] = @belapsed begin
-            y, _ = $(model_compiled)($(x), $(ps_ra), $(st_ra))
-            Reactant.synchronize(y)
-        end
+        timings[b] = @belapsed $(model_compiled)($(x), $(ps_ra), $(st_ra))
 
         println("Best timing: $(timings[b]) s")
     end