lattice/latticegpu.jl
1
0
Fork 0
mirror of https://igit.ific.uv.es/alramos/latticegpu.jl.git synced 2025-05-14 11:13:42 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added SF_bndfix! function

Browse source
This commit is contained in:
Fernando P. Panadero 2023-11-20 14:44:30 +01:00
parent dea04bccff
commit 32c4c84864
2 changed files with 22 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/Dirac/Dirac.jl
View file

@ -271,6 +271,26 @@ function DwdagDw!(so, U, si, dpar::DiracParam, dws::DiracWorkspace, lp::Union{Sp
end
export Dw!, g5Dw!, DwdagDw!
function SF_bndfix!(sp, lp::Union{SpaceParm{4,6,BC_SF_ORBI,D},SpaceParm{4,6,BC_SF_AFWB,D}}) where {D}
CUDA.@sync begin
CUDA.@cuda threads=lp.bsz blocks=lp.rsz krnl_sfbndfix!(sp, lp)
end
return nothing
end
function krnl_sfbndfix!(sp,lp::SpaceParm)
b=Int64(CUDA.threadIdx().x)
r=Int64(CUDA.blockIdx().x)
if (point_time((b,r),lp) == 1)
sp[b,r] = 0.0*sp[b,r]
end
return nothing
end
export Dw!, g5Dw!, DwdagDw!, SF_bndfix!
end

2
src/LatticeGPU.jl
View file

@ -57,7 +57,7 @@ export pmul, gpmul, gdagpmul, dmul
include("Dirac/Dirac.jl")
using .Dirac
export DiracWorkspace, DiracParam
export Dw!, g5Dw!, DwdagDw!
export Dw!, g5Dw!, DwdagDw!, SF_bndfix!
include("Solvers/Solvers.jl")