From 32c4c84864f53e81149b7efb081a213e2c4a0272 Mon Sep 17 00:00:00 2001
From: "Fernando P. Panadero" <fernando.p@csic.es>
Date: Mon, 20 Nov 2023 14:44:30 +0100
Subject: [PATCH] Added SF_bndfix! function

---
 src/Dirac/Dirac.jl | 22 +++++++++++++++++++++-
 src/LatticeGPU.jl  |  2 +-
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/src/Dirac/Dirac.jl b/src/Dirac/Dirac.jl
index d6e1f29..dbd25fb 100644
--- a/src/Dirac/Dirac.jl
+++ b/src/Dirac/Dirac.jl
@@ -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
diff --git a/src/LatticeGPU.jl b/src/LatticeGPU.jl
index 40e4544..f7e5e84 100644
--- a/src/LatticeGPU.jl
+++ b/src/LatticeGPU.jl
@@ -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")