lattice/latticegpu.jl
1
0
Fork 0
mirror of https://igit.ific.uv.es/alramos/latticegpu.jl.git synced 2025-07-01 14:29:26 +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

6
src/YM/YMfields.jl
View file

@ -36,7 +36,7 @@ end
function krnl_assign_SU3!(frc::AbstractArray{T}, m, lp::SpaceParm{N,M,BC_PERIODIC,D}) where {T,N,M,D}
b, r = CUDA.threadIdx().x, CUDA.blockIdx().x
b, r = assign_thx()
for id in 1:lp.ndim
frc[b,id,r] = SU3alg(m[b,id,1,r], m[b,id,2,r], m[b,id,3,r],
m[b,id,4,r], m[b,id,5,r], m[b,id,6,r],
@ -47,7 +47,7 @@ end
function krnl_assign_SU3!(frc::AbstractArray{T}, m, 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)
if ((B==BC_SF_AFWB)||(B==BC_SF_ORBI))
@ -72,7 +72,7 @@ end
function krnl_assign_SU2!(frc, m, lp::SpaceParm{N,M,BC_PERIODIC,D}) where {N,M,D}
b, r = CUDA.threadIdx().x, CUDA.blockIdx().x
b, r = assign_thx()
for id in 1:lp.ndim
frc[b,id,r] = SU2alg(m[b,id,1,r], m[b,id,2,r], m[b,id,3,r])
end