Skip to content

Commit 43a205d

Browse files
committed
Update of dagger-based cholesky bechmark
1 parent 6e7f8e3 commit 43a205d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

examples/agentic/generate-dagger-linear-solver/cholesky_benchmark.jl

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,10 +66,13 @@ end # 1.046506 seconds (88.80 k allocations: 4.146 MiB, 7.74% gc time, 18 lock
6666
# end samples = 1
6767

6868
# Relative error
69-
e_d = norm(A_d*x_d - b_d)/norm(b_d)
69+
e_d = Dagger.with_options(scope=Dagger.scope(;cuda_gpu=1)) do
70+
norm(A_d*x_d - b_d)/norm(b_d)
71+
end
7072

7173
# Free memory
7274
A_d = nothing
7375
b_d = nothing
76+
x_d = nothing
7477
GC.gc()
7578

0 commit comments

Comments
 (0)