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

Added option to save memory

Browse source
This commit is contained in:
Alberto Ramos 2021-09-24 15:49:53 +02:00
parent 0587e5ffea
commit 879ca7dab4
5 changed files with 152 additions and 153 deletions
Download patch file Download diff file Expand all files Collapse all files

31
src/YM/YM.jl
View file

@ -29,33 +29,44 @@ struct YMworkspace{T}
PRC
frc1
frc2
fpln
mom
U1
cm # complex of volume
rm # float of volume
function YMworkspace(::Type{G}, ::Type{T}, lp::SpaceParm) where {G <: Group, T <: AbstractFloat}
function YMworkspace(::Type{G}, ::Type{T}, lp::SpaceParm; save_mem = false) 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)
f1 = field(SU2alg{T}, lp)
f2 = field(SU2alg{T}, lp)
if save_mem
fp = nothing
else
fp = field_pln(SU2alg{T}, lp)
end
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)
f1 = field(SU3alg{T}, lp)
f2 = field(SU3alg{T}, lp)
if save_mem
fp = nothing
else
fp = field_pln(SU2alg{T}, lp)
end
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)
return new{T}(GRP,ALG,T,f1, f2, fp, mm, u1, cs, rs)
end
end
export YMworkspace