mirror of
https://igit.ific.uv.es/alramos/latticegpu.jl.git
synced 2025-05-14 11:13:42 +02:00
72 lines
1.6 KiB
Julia
72 lines
1.6 KiB
Julia
###
|
|
### "THE BEER-WARE LICENSE":
|
|
### Alberto Ramos wrote this file. As long as you retain this
|
|
### notice you can do whatever you want with this stuff. If we meet some
|
|
### day, and you think this stuff is worth it, you can buy me a beer in
|
|
### return. <alberto.ramos@cern.ch>
|
|
###
|
|
### file: YM.jl
|
|
### created: Mon Jul 12 16:23:51 2021
|
|
###
|
|
|
|
|
|
module YM
|
|
|
|
using CUDA, Random, StructArrays
|
|
using ..Space
|
|
using ..Groups
|
|
|
|
struct GaugeParm{T}
|
|
beta::T
|
|
cG::NTuple{2,T}
|
|
ng::Int64
|
|
end
|
|
export GaugeParm
|
|
|
|
struct YMworkspace{T}
|
|
GRP
|
|
ALG
|
|
PRC
|
|
frc1
|
|
frc2
|
|
mom
|
|
U1
|
|
cm # complex of volume
|
|
rm # float of volume
|
|
function YMworkspace(::Type{G}, ::Type{T}, lp::SpaceParm) where {G <: Group, T <: AbstractFloat}
|
|
|
|
if (G == SU2)
|
|
GRP = SU2
|
|
ALG = SU2alg
|
|
f1 = field(SU2alg, T, lp)
|
|
f2 = field(SU2alg, T, lp)
|
|
mm = field(SU2alg, T, lp)
|
|
u1 = field(SU2, T, lp)
|
|
end
|
|
|
|
if (G == SU3)
|
|
GRP = SU3
|
|
ALG = SU3alg
|
|
f1 = field(SU3alg, T, lp)
|
|
f2 = field(SU3alg, T, lp)
|
|
mm = field(SU3alg, T, lp)
|
|
u1 = field(SU3, T, lp)
|
|
end
|
|
cs = CuArray{Complex{T}, 2}(undef, lp.bsz,lp.rsz)
|
|
rs = CuArray{T, 2}(undef, lp.bsz,lp.rsz)
|
|
|
|
return new{T}(GRP,ALG,T,f1, f2, mm, u1, cs, rs)
|
|
end
|
|
end
|
|
export YMworkspace
|
|
|
|
include("YMfields.jl")
|
|
export field, field_pln, randomize!, zero!, norm2
|
|
|
|
include("YMact.jl")
|
|
export krnl_plaq!, force0_wilson!
|
|
|
|
include("YMhmc.jl")
|
|
export gauge_action, hamiltonian, plaquette, HMC!, OMF4!
|
|
|
|
end
|