Used T in pfrandomize

2ba57494 · Fernando Pérez Panadero · 1110a1e7 · 2ba57494
Commit 2ba57494 authored Oct 19, 2023 by Fernando Pérez Panadero
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

src/Dirac/Dirac.jl src/Dirac/Dirac.jl +3 -3

No files found.
--- a/src/Dirac/Dirac.jl
+++ b/src/Dirac/Dirac.jl
@@ -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
-            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.@cuda threads=lp.bsz blocks=lp.rsz krnl_assign_pf!(f,p,lp,t)
            end