Default to Float32 for grids on GPU?
navidcy opened this issue · 0 comments
navidcy commented
@glwagner should we change
FourierFlows.jl/src/CuFourierFlows.jl
Lines 5 to 7 in 796ed2d
to
OneDGrid(dev::GPU, args...; kwargs...) = OneDGrid(args...; ArrayType=CuArray, T=Float32, kwargs...)
TwoDGrid(dev::GPU, args...; kwargs...) = TwoDGrid(args...; ArrayType=CuArray, T=Float32, kwargs...)
ThreeDGrid(dev::GPU, args...; kwargs...) = ThreeDGrid(args...; ArrayType=CuArray, T=Float32, kwargs...)
?