Update ext/ReactantExt/precompile_workloads.jl

avik-pal · github-actions[bot] · web-flow · commit 7cb31e8f3ae3 · 2026-01-21T20:56:44.000-05:00
Co-authored-by: github-actions[bot] &lt;41898282+github-actions[bot]@users.noreply.github.com&gt;
diff --git a/ext/ReactantExt/precompile_workloads.jl b/ext/ReactantExt/precompile_workloads.jl
@@ -38,7 +38,8 @@ if Reactant.Reactant_jll.is_available()
                 Lux.FlattenLayer(),
                 Lux.Dense(64 => 10),
             )
-            ps_conv_model, st_conv_model = Lux.setup(Random.default_rng(), conv_model) |> dev
+            ps_conv_model, st_conv_model =
+                Lux.setup(Random.default_rng(), conv_model) |> dev
 
             x = rand(Float32, (28, 28, 3, 2)) |> dev
 

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,8 @@ if Reactant.Reactant_jll.is_available()`
`38`	`38`	`Lux.FlattenLayer(),`
`39`	`39`	`Lux.Dense(64 => 10),`
`40`	`40`	`)`
`41`		`- ps_conv_model, st_conv_model = Lux.setup(Random.default_rng(), conv_model) \|> dev`
	`41`	`+ ps_conv_model, st_conv_model =`
	`42`	`+ Lux.setup(Random.default_rng(), conv_model) \|> dev`
`42`	`43`
`43`	`44`	`x = rand(Float32, (28, 28, 3, 2)) \|> dev`
`44`	`45`