lattice/latticegpu.jl
1
0
Fork 0
mirror of https://igit.ific.uv.es/alramos/latticegpu.jl.git synced 2025-07-02 06:49:27 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Wrap threadIdx, etc in an inline function

Browse source
This commit is contained in:
Alberto Ramos 2022-01-05 20:04:55 +01:00
parent 65a619e554
commit 584b92edb0
7 changed files with 21 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/YM/YMact.jl
View file

@ -11,7 +11,7 @@
function krnl_impr!(plx, U::AbstractArray{T}, c0, c1, Ubnd::NTuple{NB,T}, cG, ztw, lp::SpaceParm{N,M,B,D}) where {T,NB,N,M,B,D}
b, r = CUDA.threadIdx().x, CUDA.blockIdx().x
b, r = assign_thx()
it = point_time((b, r), lp)
Ush = @cuStaticSharedMem(T, (D,2))
@ -121,7 +121,7 @@ end
function krnl_plaq!(plx, U::AbstractArray{T}, Ubnd, cG, ztw, lp::SpaceParm{N,M,B,D}) where {T,N,M,B,D}
b, r = CUDA.threadIdx().x, CUDA.blockIdx().x
b, r = assign_thx()
it = point_time((b, r), lp)
Ush = @cuStaticSharedMem(T, (D,2))
@ -172,7 +172,7 @@ end
function krnl_force_wilson_pln!(frc1, frc2, U::AbstractArray{T}, Ubnd, cG, ztw, ipl, lp::SpaceParm{N,M,B,D}) where {T,N,M,B,D}
b, r = CUDA.threadIdx().x, CUDA.blockIdx().x
b, r = assign_thx()
it = point_time((b,r), lp)
Ush = @cuStaticSharedMem(T, (D,2))
@ -233,7 +233,7 @@ end
function krnl_force_impr_pln!(frc1, frc2, U::AbstractArray{T}, c0, c1, Ubnd, cG, ztw, ipl, lp::SpaceParm{N,M,B,D}) where {T,N,M,B,D}
b, r = CUDA.threadIdx().x, CUDA.blockIdx().x
b, r = assign_thx()
it = point_time((b, r), lp)
Ush = @cuStaticSharedMem(T, (D,2))