Used T in pfrandomize

parent 1110a1e7
...@@ -62,7 +62,7 @@ struct DiracWorkspace{T} ...@@ -62,7 +62,7 @@ struct DiracWorkspace{T}
csw = tensor_field(U3alg{T},lp) csw = tensor_field(U3alg{T},lp)
cs = scalar_field_point(Complex{Float64}, lp) cs = scalar_field_point(Complex{Float64}, lp)
return new{T}(sr,sp,sAp,st,csw,cs) return new{T}(sr,sp,sAp,st,csw,cs)
end end
end end
...@@ -439,10 +439,10 @@ end ...@@ -439,10 +439,10 @@ end
function pfrandomize!(f,lp::SpaceParm,t::Int64=0) where {T} function pfrandomize!(f::AbstractArray{T},lp::SpaceParm,t::Int64=0) where {T}
@timeit "Randomize pseudofermion field" begin @timeit "Randomize pseudofermion field" begin
p = ntuple(i->CUDA.randn(Float64, lp.bsz, 3, lp.rsz,2),4) # complex generation not suported for Julia 1.5.4 p = ntuple(i->CUDA.randn(T, lp.bsz, 3, lp.rsz,2),4) # complex generation not suported for Julia 1.5.4
CUDA.@sync begin CUDA.@sync begin
CUDA.@cuda threads=lp.bsz blocks=lp.rsz krnl_assign_pf!(f,p,lp,t) CUDA.@cuda threads=lp.bsz blocks=lp.rsz krnl_assign_pf!(f,p,lp,t)
end end
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment