First try to properly add improvement

src/Fields/Fields.jl

@@ -18,6 +18,7 @@ using ..Space
 vector_field(::Type{T}, lp::SpaceParm)     where {T} = CuArray{T, 3}(undef, lp.bsz, lp.ndim, lp.rsz)
 scalar_field(::Type{T}, lp::SpaceParm)     where {T} = CuArray{T, 2}(undef, lp.bsz, lp.rsz)
 nscalar_field(::Type{T}, n, lp::SpaceParm) where {T} = CuArray{T, 3}(undef, lp.bsz, n, lp.rsz)
+tensor_field(::Type{T}, lp::SpaceParm)     where {T} = CuArray{T, 3}(undef, lp.bsz, lp.npls, lp.rsz)
 
 scalar_field_point(::Type{T}, lp::SpaceParm{N,M,D}) where {T,N,M,D} = CuArray{T, N}(undef, lp.iL...)
 

src/Solvers/CG.jl

@@ -34,7 +34,7 @@ return sum(sumf)
 end
 
 
-function CG!(si, U, A, dpar::DiracParam, lp::SpaceParm, dws::DiracWorkspace{T}, ymws = "ymws", gp = "gp", maxiter::Int64 = 10, tol=1.0) where {T}
+function CG!(si, U, A, dpar::DiracParam, lp::SpaceParm, dws::DiracWorkspace{T}, ymws::YMWorkspace, maxiter::Int64 = 10, tol=1.0) where {T}
 
     dws.sr .= si
     dws.sp .= si

src/YM/YM.jl

@@ -83,6 +83,7 @@ struct YMworkspace{T}
                 f2 = vector_field(SU2alg{T}, lp)
                 mm = vector_field(SU2alg{T}, lp)
                 u1 = vector_field(SU2{T},    lp)
+                csw = tensor_field(SU2alg{T},lp)
             end
             
             if (G == SU3)
@@ -92,6 +93,7 @@ struct YMworkspace{T}
                 f2 = vector_field(SU3alg{T}, lp)
                 mm = vector_field(SU3alg{T}, lp)
                 u1 = vector_field(SU3{T},    lp)
+                csw = tensor_field(SU3alg{T},lp)
             end
             cs = scalar_field_point(Complex{T}, lp)
             rs = scalar_field_point(T, lp)