made vertical grid equality work for GPUs

CliMA · Oct 18, 2021 · 1cea014 · 1cea014
1 parent 3b934b5
commit 1cea014
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/OutputWriters/checkpointer.jl b/src/OutputWriters/checkpointer.jl
@@ -194,7 +194,7 @@ function set!(model, filepath::AbstractString)
 
         # Validate the grid
         checkpointed_grid = file["grid"]
-        model.grid == checkpointed_grid ||
+        CUDA.@allowscalar model.grid == checkpointed_grid ||
             error("The grid associated with $filepath and model.grid are not the same!")
 
         # Set model fields and tendency fields