Skip to content

Commit d85aca5

Browse files
committed
chore: run the formatter
1 parent f724d7e commit d85aca5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

benchmark/vit.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,11 @@ dev = gpu_device()
99
model = Vision.ViT(:tiny);
1010
ps, st = Lux.setup(Random.default_rng(), model);
1111

12-
ps_gpu, st_gpu = (ps, st) |> dev;
12+
ps_gpu = dev(ps);
13+
st_gpu = dev(st);
1314

1415
x = rand(Float32, 256, 256, 3, 4);
15-
x_gpu = x |> dev;
16+
x_gpu = dev(x);
1617

1718
lux_timing = @benchmark begin
1819
Lux.apply($model, $x_gpu, $ps_gpu, $st_gpu)

0 commit comments

Comments
 (0)