proper streamfunction initialization

FourierFlows · Jul 7, 2022 · d91872a · d91872a
1 parent 7ef9de8
commit d91872a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_multilayerqg.jl b/test/test_multilayerqg.jl
@@ -357,8 +357,8 @@ function test_mqg_energies(dev::Device=CPU();
 
   ψ = zeros(dev, eltype(gr), (gr.nx, gr.ny, nlayers))
 
-  CUDA.@allowscalar @views ψ[:, :, 1] .=       cos(2k₀ * x) * cos(2l₀ * y)
-  CUDA.@allowscalar @views ψ[:, :, 2] .= 1/2 * cos(2k₀ * x) * cos(2l₀ * y)
+  CUDA.@allowscalar @views @. ψ[:, :, 1] =       cos(2k₀ * x) * cos(2l₀ * y)
+  CUDA.@allowscalar @views @. ψ[:, :, 2] = 1/2 * cos(2k₀ * x) * cos(2l₀ * y)
 
   MultiLayerQG.set_ψ!(prob, ψ)