Merge pull request #107 from FourierFlows/small-fix

Small fix -- adds CUDA.@allowscalar in enstrophy_dissipation()
FourierFlows · Sep 11, 2020 · 5f7795c · 5f7795c
2 parents e8c39e4 + 0c1fb24
commit 5f7795c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/twodnavierstokes.jl b/src/twodnavierstokes.jl
@@ -284,7 +284,7 @@ Returns the domain-averaged dissipation rate of enstrophy. nν must be >= 1.
 @inline function enstrophy_dissipation(prob)
   sol, vars, params, grid = prob.sol, prob.vars, prob.params, prob.grid
   @. vars.uh = grid.Krsq^params.nν * abs2(sol)
-  vars.uh[1, 1] = 0
+  CUDA.@allowscalar vars.uh[1, 1] = 0
   return params.ν / (grid.Lx * grid.Ly) * parsevalsum(vars.uh, grid)
 end