mirror of
https://igit.ific.uv.es/alramos/latticegpu.jl.git
synced 2025-05-14 11:13:42 +02:00
54 lines
1.2 KiB
Julia
54 lines
1.2 KiB
Julia
using CUDA, LatticeGPU, TimerOutputs
|
|
|
|
#Check that Dw ( (DwdagDw)^{-1} g5 Dw g5 ) psi = psi for random fields
|
|
|
|
@timeit "Rand solver test" begin
|
|
|
|
@timeit "Generate random fields" begin
|
|
|
|
lp = SpaceParm{4}((16,16,16,16), (4,4,4,4), 0, (0,0,0,0,0,0))
|
|
gp = GaugeParm{Float64}(SU3{Float64}, 6.0, 1.0)
|
|
ymws = YMworkspace(SU3, Float64, lp)
|
|
dpar = DiracParam{Float64}(SU3fund,2.3,0.0,(1.0,1.0,1.0,1.0),0.0)
|
|
dws = DiracWorkspace(SU3fund{Float64},Float64,lp);
|
|
|
|
randomize!(ymws.mom, lp, ymws)
|
|
U = exp.(ymws.mom)
|
|
|
|
rpsi = scalar_field(Spinor{4,SU3fund{Float64}},lp)
|
|
pfrandomize!(rpsi,lp)
|
|
|
|
end
|
|
|
|
prop = scalar_field(Spinor{4,SU3fund{Float64}},lp)
|
|
|
|
function krnlg5!(src)
|
|
b=Int64(CUDA.threadIdx().x)
|
|
r=Int64(CUDA.blockIdx().x)
|
|
src[b,r] = dmul(Gamma{5},src[b,r])
|
|
return nothing
|
|
end
|
|
|
|
CUDA.@sync begin
|
|
CUDA.@cuda threads=lp.bsz blocks=lp.rsz krnlg5!(rpsi)
|
|
end
|
|
g5Dw!(prop,U,rpsi,dpar,dws,lp)
|
|
CG!(prop,U,DwdagDw!,dpar,lp,dws,10000,1.0e-14)
|
|
|
|
Dw!(dws.sp,U,prop,dpar,dws,lp)
|
|
|
|
CUDA.@sync begin
|
|
CUDA.@cuda threads=lp.bsz blocks=lp.rsz krnlg5!(rpsi)
|
|
end
|
|
|
|
res = sum(norm2.(rpsi-dws.sp))
|
|
|
|
|
|
if res < 1.0e-6
|
|
print("Drand test passed with ",res,"% error!\n")
|
|
|
|
else
|
|
error("Drand test failed with difference: ",res,"\n")
|
|
end
|
|
|
|
end
|