Solver modified to include the improved action

src/Solvers/CG.jl

@@ -34,7 +34,7 @@ return sum(sumf)
 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.sp .= si
@@ -47,7 +47,7 @@ function CG!(si, U, A, dpar::DiracParam, lp::SpaceParm, dws::DiracWorkspace{T},
 
     niter = 0
     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  = CUDA.mapreduce(x -> dot(x[1],x[2]), +, zip(dws.sp, dws.sAp))