mirror of
https://igit.ific.uv.es/alramos/latticegpu.jl.git
synced 2025-05-14 19:23:42 +02:00
107 lines
2.6 KiB
Julia
107 lines
2.6 KiB
Julia
###
|
|
### "THE BEER-WARE LICENSE":
|
|
### Alberto Ramos and Carlos Pena 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 us a beer in
|
|
### return. <alberto.ramos@cern.ch> <carlos.pena@uam.es>
|
|
###
|
|
### file: Dirac.jl
|
|
### created: Thu Nov 18 17:20:24 2021
|
|
###
|
|
|
|
|
|
module Dirac
|
|
|
|
using CUDA, TimerOutputs
|
|
using ..Space
|
|
using ..Groups
|
|
using ..Fields
|
|
using ..YM
|
|
using ..Spinors
|
|
|
|
struct DiracWorkspace{T}
|
|
sr
|
|
sp
|
|
sAp
|
|
st
|
|
|
|
function DiracWorkspace(::Type{G}, ::Type{T}, lp::SpaceParm{4,6,B,D}) where {G,T <: AbstractFloat, B,D}
|
|
|
|
sr = scalar_field(Spinor{4,G}, lp)
|
|
sp = scalar_field(Spinor{4,G}, lp)
|
|
sAp = scalar_field(Spinor{4,G}, lp)
|
|
st = scalar_field(Spinor{4,G}, lp)
|
|
return new{T}(sr,sp,sAp,st)
|
|
end
|
|
end
|
|
export DiracWorkspace
|
|
|
|
function Dw!(so, U, si, m0, lp::SpaceParm)
|
|
|
|
@timeit "Dw" begin
|
|
CUDA.@sync begin
|
|
CUDA.@cuda threads=lp.bsz blocks=lp.rsz krnl_Dw!(so, U, si, m0, [1,1,1,1], lp)
|
|
end
|
|
end
|
|
|
|
return nothing
|
|
end
|
|
|
|
function DwdagDw!(so, U, si, m0, st, lp::SpaceParm)
|
|
|
|
@timeit "DwdagDw" begin
|
|
CUDA.@sync begin
|
|
CUDA.@cuda threads=lp.bsz blocks=lp.rsz krnl_g5Dw!(st, U, si, m0, [1,1,1,1], lp)
|
|
end
|
|
CUDA.@sync begin
|
|
CUDA.@cuda threads=lp.bsz blocks=lp.rsz krnl_g5Dw!(so, U, st, m0, [1,1,1,1], lp)
|
|
end
|
|
end
|
|
|
|
return nothing
|
|
end
|
|
|
|
|
|
function krnl_Dw!(so, U, si, m0, th, lp::SpaceParm{4,6,B,D}) where {B,D}
|
|
|
|
b, r = assign_thx()
|
|
|
|
# For SF:
|
|
# - cttilde affects mass term at x0 = a, T-a
|
|
# - Spinor can be periodic as long as 0 at x_0=0
|
|
@inbounds begin
|
|
so[b,r] = (4+m0)*si[b,r]
|
|
for id in 1:4
|
|
bu, ru = up((b,r), id, lp)
|
|
bd, rd = dw((b,r), id, lp)
|
|
|
|
so[b,r] -= ( th[id]*gpmul(Pgamma{id,-1},U[b,id,r],si[bu,ru]) +
|
|
conj(th[id])*gdagpmul(Pgamma{id,+1},U[bd,id,rd],si[bd,rd]) )/2
|
|
end
|
|
end
|
|
|
|
return nothing
|
|
end
|
|
|
|
function krnl_g5Dw!(so, U, si, m0, th, lp::SpaceParm{4,6,B,D}) where {B,D}
|
|
|
|
b, r = assign_thx()
|
|
|
|
@inbounds begin
|
|
so[b,r] = (4+m0)*si[b,r]
|
|
for id in 1:4
|
|
bu, ru = up((b,r), id, lp)
|
|
bd, rd = dw((b,r), id, lp)
|
|
|
|
so[b,r] -= ( th[id]*gpmul(Pgamma{id,-1},U[b,id,r],si[bu,ru]) +
|
|
conj(th[id])*gdagpmul(Pgamma{id,+1},U[bd,id,rd],si[bd,rd]) )/2
|
|
end
|
|
so[b,r] = dmul(Gamma{5}, so[b,r])
|
|
end
|
|
|
|
return nothing
|
|
end
|
|
|
|
export Dw!, DwdagDw!
|
|
|
|
end
|