Skip to content

L
LGPU FPP
Commit c4fa8e78 authored by Fernando Pérez Panadero's avatar Fernando Pérez Panadero
Browse files
Options

Solver modified to include the improved action

parent a7d23188
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/Solvers/CG.jl
View file @ c4fa8e78
...@@ -34,7 +34,7 @@ return sum(sumf) ...@@ -34,7 +34,7 @@ return sum(sumf)
end end
function CG!(si, U, A, dpar::DiracParam, lp::SpaceParm, dws::DiracWorkspace{T}, maxiter::Int64 = 10, tol=1.0) where {T} function CG!(si, U, A, dpar::DiracParam, lp::SpaceParm, dws::DiracWorkspace{T}, ymws = "ymws", gp = "gp", maxiter::Int64 = 10, tol=1.0) where {T}
dws.sr .= si dws.sr .= si
dws.sp .= si dws.sp .= si
...@@ -47,7 +47,7 @@ function CG!(si, U, A, dpar::DiracParam, lp::SpaceParm, dws::DiracWorkspace{T}, ...@@ -47,7 +47,7 @@ function CG!(si, U, A, dpar::DiracParam, lp::SpaceParm, dws::DiracWorkspace{T},
niter = 0 niter = 0
for i in 1:maxiter for i in 1:maxiter
A(dws.sAp, U, dws.sp, dpar, dws.st, lp) A(dws.sAp, U, dws.sp, dpar, dws.st, lp, ymws, gp)
prod = field_dot(dws.sp,dws.sAp,sumf,lp) prod = field_dot(dws.sp,dws.sAp,sumf,lp)
#prod = CUDA.mapreduce(x -> dot(x[1],x[2]), +, zip(dws.sp, dws.sAp)) #prod = CUDA.mapreduce(x -> dot(x[1],x[2]), +, zip(dws.sp, dws.sAp))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment