but what if 1D x is array?

FourierFlows · Mar 6, 2021 · 435c998 · 435c998
1 parent bfadb0f
commit 435c998
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils.jl b/src/utils.jl
@@ -248,7 +248,7 @@ end
 Returns an array, of the ArrayType of the device `grid` lives on, that contains the values of
 function `func` evaluated on the `grid`.
 """
-on_grid(func, grid::OneDGrid) = func.(grid.x)
+on_grid(func, grid::OneDGrid) = CUDA.@allowscalar func.(grid.x)
 
 function on_grid(func, grid::TwoDGrid)
   x, y = gridpoints(grid)