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

Working version. Comparison with one field seem ok.

Browse source
This commit is contained in:
Alberto Ramos 2021-10-15 15:52:21 +02:00
parent 30baeade9f
commit f0ca0bbde1
6 changed files with 85 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

120
src/Scalar/ScalarHMC.jl
View file

@ -12,78 +12,80 @@
function hamiltonian(mom, U, pmom, Phi, lp, gp, sp, ymws)
SG = gauge_action(U, lp, gp, ymws)
SS = scalar_action(U, Phi, lp, sp, ymws)
PG = CUDA.mapreduce(norm2, +, mom)/2
PS = CUDA.mapreduce(norm2, +, pmom)/2
println("Hamiltonian: ", SG, " ", SS, " ", PG, " ",PS)
@timeit "Computing Hamiltonian" begin
SG = gauge_action(U, lp, gp, ymws)
SS = scalar_action(U, Phi, lp, sp, ymws)
PG = CUDA.mapreduce(norm2, +, mom)/2
PS = CUDA.mapreduce(norm2, +, pmom)/2
end
return SG+SS+PG+PS
end
function HMC!(U, Phi, eps, ns, lp::SpaceParm, gp::GaugeParm, sp::ScalarParm, ymws::YMworkspace{T}, sws::ScalarWorkspace; noacc=false) where T
int = omf4(T, eps, ns)
ymws.U1 .= U
sws.Phi .= Phi
randomize!(ymws.mom, lp, ymws)
randomize!(sws.mom, sp, lp, ymws)
hini = hamiltonian(ymws.mom, U, sws.mom, Phi, lp, gp, sp, ymws)
println(hini)
MD!(ymws.mom, U, sws.mom, Phi, int, lp, gp, sp, ymws, sws)
dh = hamiltonian(ymws.mom, U, sws.mom, Phi, lp, gp, sp, ymws) - hini
println(dh+hini)
pacc = exp(-dh)
acc = true
if (noacc)
return dh, acc
end
if (pacc < 1.0)
r = rand()
if (pacc < r)
U .= ymws.U1
Phi .= sws.Phi
acc = false
@timeit "HMC trajectory" begin
int = omf4(T, eps, ns)
ymws.U1 .= U
sws.Phi .= Phi
randomize!(ymws.mom, lp, ymws)
randomize!(sws.mom, sp, lp, ymws)
hini = hamiltonian(ymws.mom, U, sws.mom, Phi, lp, gp, sp, ymws)
MD!(ymws.mom, U, sws.mom, Phi, int, lp, gp, sp, ymws, sws)
dh = hamiltonian(ymws.mom, U, sws.mom, Phi, lp, gp, sp, ymws) - hini
pacc = exp(-dh)
acc = true
if (noacc)
return dh, acc
end
if (pacc < 1.0)
r = rand()
if (pacc < r)
U .= ymws.U1
Phi .= sws.Phi
acc = false
end
end
end
return dh, acc
end
function MD!(mom, U, pmom, Phi, int::IntrScheme{NI, T}, lp::SpaceParm, gp::GaugeParm{T}, sp::ScalarParm, ymws::YMworkspace{T}, sws::ScalarWorkspace) where {NI, T <: AbstractFloat}
YM.force_gauge(ymws, U, gp.c0, lp)
force_scalar(ymws, sws, U, Phi, sp, gp, lp)
mom .= mom .+ (int.r[1]*int.eps) .* ymws.frc1
pmom .= pmom .+ (int.r[1]*int.eps) .* sws.frc1
for i in 1:int.ns
k = 2
off = 1
for j in 1:NI-1
U .= expm.(U, mom, int.eps*int.r[k])
Phi .= Phi .+ (int.eps*int.r[k]).*pmom
if k == NI
off = -1
@timeit "MD evolution" begin
YM.force_gauge(ymws, U, gp.c0, lp)
force_scalar(ymws, sws, U, Phi, sp, gp, lp)
mom .= mom .+ (int.r[1]*int.eps) .* ymws.frc1
pmom .= pmom .+ (int.r[1]*int.eps) .* sws.frc1
for i in 1:int.ns
k = 2
off = 1
for j in 1:NI-1
U .= expm.(U, mom, int.eps*int.r[k])
Phi .= Phi .+ (int.eps*int.r[k]).*pmom
if k == NI
off = -1
end
k += off
YM.force_gauge(ymws, U, gp.c0, lp)
force_scalar(ymws, sws, U, Phi, sp, gp, lp)
if (i < int.ns) && (k == 1)
mom .= mom .+ (2*int.r[k]*int.eps) .* ymws.frc1
pmom .= pmom .+ (2*int.r[k]*int.eps) .* sws.frc1
else
mom .= mom .+ (int.r[k]*int.eps) .* ymws.frc1
pmom .= pmom .+ (int.r[k]*int.eps) .* sws.frc1
end
k += off
end
k += off
YM.force_gauge(ymws, U, gp.c0, lp)
force_scalar(ymws, sws, U, Phi, sp, gp, lp)
if (i < int.ns) && (k == 1)
mom .= mom .+ (2*int.r[k]*int.eps) .* ymws.frc1
pmom .= pmom .+ (2*int.r[k]*int.eps) .* sws.frc1
else
mom .= mom .+ (int.r[k]*int.eps) .* ymws.frc1
pmom .= pmom .+ (int.r[k]*int.eps) .* sws.frc1
end
k += off
end
end