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

Working SU(3) code

Browse source
This commit is contained in:
Alberto Ramos 2021-07-19 02:03:31 +02:00
parent c341cd24aa
commit f6be70070e
2 changed files with 237 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

112
src/YM/YMfields.jl Normal file
View file

@ -0,0 +1,112 @@
###
### "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: YMfields.jl
### created: Thu Jul 15 15:16:47 2021
###
function field(::Type{T}, lp::SpaceParm) where {T <: Union{Group, Algebra}}
sz = lp.iL..., lp.ndim
if (T == SU2)
As = StructArray{SU2}((ones(ComplexF64, sz), zeros(ComplexF64, sz)))
elseif (T == SU2alg)
As = StructArray{SU2alg}((zeros(Float64, sz),
zeros(Float64, sz),
zeros(Float64, sz)))
elseif (T == SU3)
As = StructArray{SU3}((ones(ComplexF64, sz), zeros(ComplexF64, sz), zeros(ComplexF64, sz), zeros(ComplexF64, sz), ones(ComplexF64, sz), zeros(ComplexF64, sz)))
elseif (T == SU3alg)
As = StructArray{SU3alg}((zeros(Float64, sz),
zeros(Float64, sz),
zeros(Float64, sz),
zeros(Float64, sz),
zeros(Float64, sz),
zeros(Float64, sz),
zeros(Float64, sz),
zeros(Float64, sz)))
end
return replace_storage(CuArray, As)
end
function randomn!(X)
if (eltype(X) == SU2alg)
randn!(CURAND.default_rng(), LazyRows(X).t1)
randn!(CURAND.default_rng(), LazyRows(X).t2)
randn!(CURAND.default_rng(), LazyRows(X).t3)
elseif (eltype(X) == SU3alg)
randn!(CURAND.default_rng(), LazyRows(X).t1)
randn!(CURAND.default_rng(), LazyRows(X).t2)
randn!(CURAND.default_rng(), LazyRows(X).t3)
randn!(CURAND.default_rng(), LazyRows(X).t4)
randn!(CURAND.default_rng(), LazyRows(X).t5)
randn!(CURAND.default_rng(), LazyRows(X).t6)
randn!(CURAND.default_rng(), LazyRows(X).t7)
randn!(CURAND.default_rng(), LazyRows(X).t8)
end
return nothing
end
function zero!(X)
if (eltype(X) == SU2alg)
fill!(LazyRows(X).t1, 0.0)
fill!(LazyRows(X).t2, 0.0)
fill!(LazyRows(X).t3, 0.0)
end
if (eltype(X) == SU3alg)
fill!(LazyRows(X).t1, 0.0)
fill!(LazyRows(X).t2, 0.0)
fill!(LazyRows(X).t3, 0.0)
fill!(LazyRows(X).t4, 0.0)
fill!(LazyRows(X).t5, 0.0)
fill!(LazyRows(X).t6, 0.0)
fill!(LazyRows(X).t7, 0.0)
fill!(LazyRows(X).t8, 0.0)
end
if (eltype(X) == SU2)
fill!(LazyRows(X).t1, complex(1.0))
fill!(LazyRows(X).t2, complex(0.0))
end
if (eltype(X) == SU3)
fill!(LazyRows(X).u11, complex(1.0))
fill!(LazyRows(X).u12, complex(0.0))
fill!(LazyRows(X).u13, complex(0.0))
fill!(LazyRows(X).u21, complex(0.0))
fill!(LazyRows(X).u22, complex(1.0))
fill!(LazyRows(X).u23, complex(0.0))
end
return nothing
end
function norm2(X)
d = 0.0
if (eltype(X) == SU2alg)
d = CUDA.mapreduce(x->x^2, +, LazyRows(X).t1) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t2) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t3)
elseif (eltype(X) == SU2alg)
d = CUDA.mapreduce(x->x^2, +, LazyRows(X).t1) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t2) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t3) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t4) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t5) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t6) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t7) +
CUDA.mapreduce(x->x^2, +, LazyRows(X).t8)
end
return d
end