lattice/latticegpu.jl
1
0
Fork 0
mirror of https://igit.ific.uv.es/alramos/latticegpu.jl.git synced 2025-07-02 06:49:27 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Functions help

Browse source
This commit is contained in:
Fernando P. Panadero 2023-11-24 09:21:03 +01:00
parent 107586289a
commit 06e08c5016
4 changed files with 73 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/Solvers/CG.jl
View file

@ -9,11 +9,6 @@
### created: Tue Nov 30 11:10:57 2021
###
"""
function CG!
Solves the linear equation `Ax = si`
"""
function krnl_dot!(sum,fone,ftwo)
b=Int64(CUDA.threadIdx().x)
r=Int64(CUDA.blockIdx().x)
@ -32,6 +27,12 @@ function field_dot(fone::AbstractArray,ftwo::AbstractArray,sumf,lp) where {T}
return sum(sumf)
end
"""
function CG!
Solves the linear equation `Ax = si`
"""
function CG!(si, U, A, dpar::DiracParam, lp::SpaceParm, dws::DiracWorkspace{T}, maxiter::Int64 = 10, tol=1.0) where {T}
dws.sr .= si