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

Set zero

Browse source
This commit is contained in:
Alberto Ramos 2021-11-14 22:17:18 +01:00
parent 500305f76b
commit 8cd4c4fd97
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/YM/YMflow.jl
View file

@ -407,7 +407,7 @@ function krnl_add_qd!(rm, op, frc1, frc2, U, lp::SpaceParm{4,M,B,D}) where {M,B,
return nothing return nothing
end end
function krnl_field_tensor!(frc1, frc2, U::AbstractArray{T}, Ubnd, ipl1, ipl2, ztw1, ztw2, lp::SpaceParm{4,M,B,D}) where {T,M,B,D} function krnl_field_tensor!(frc1::AbstractArray{TA}, frc2, U::AbstractArray{T}, Ubnd, ipl1, ipl2, ztw1, ztw2, lp::SpaceParm{4,M,B,D}) where {TA,T,M,B,D}
b, r = CUDA.threadIdx().x, CUDA.blockIdx().x b, r = CUDA.threadIdx().x, CUDA.blockIdx().x
it = point_time((b,r), lp) it = point_time((b,r), lp)
@ -446,8 +446,8 @@ function krnl_field_tensor!(frc1, frc2, U::AbstractArray{T}, Ubnd, ipl1, ipl2, z
if SFBC && (it == lp.iL[end]) if SFBC && (it == lp.iL[end])
frc1[b,1,r] = projalg(Ush[b,1]*l1/Ush[b,2]) frc1[b,1,r] = projalg(Ush[b,1]*l1/Ush[b,2])
frc1[bu1,2,ru1] = zero(frc1[bu1,2,ru1]) frc1[bu1,2,ru1] = zero(TA)
frc1[bd,3,rd] = zero(frc1[bu1,2,ru1]) frc1[bd,3,rd] = zero(TA)
frc1[bu2,4,ru2] = projalg(l2*l1) frc1[bu2,4,ru2] = projalg(l2*l1)
else else
frc1[b,1,r] = projalg(ztw1, Ush[b,1]*l1/Ush[b,2]) frc1[b,1,r] = projalg(ztw1, Ush[b,1]*l1/Ush[b,2])
@ -487,8 +487,8 @@ function krnl_field_tensor!(frc1, frc2, U::AbstractArray{T}, Ubnd, ipl1, ipl2, z
if SFBC && (it == lp.iL[end]) if SFBC && (it == lp.iL[end])
frc2[b,1,r] = projalg(Ush[b,1]*l1/Ush[b,2]) frc2[b,1,r] = projalg(Ush[b,1]*l1/Ush[b,2])
frc1[bu1,2,ru1] = zero(frc1[bu1,2,ru1]) frc1[bu1,2,ru1] = zero(TA)
frc1[bd,3,rd] = zero(frc1[bu1,2,ru1]) frc1[bd,3,rd] = zero(TA)
frc2[bu2,4,ru2] = projalg(l2*l1) frc2[bu2,4,ru2] = projalg(l2*l1)
else else
frc2[b,1,r] = projalg(ztw1, Ush[b,1]*l1/Ush[b,2]) frc2[b,1,r] = projalg(ztw1, Ush[b,1]*l1/Ush[b,2])