add all_arrays

eth-cscs · Jan 19, 2024 · 3d5d805 · 3d5d805
1 parent c58947d
commit 3d5d805
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/shared.jl b/src/shared.jl
@@ -114,6 +114,9 @@ has_neighbor(n::Integer, dim::Integer) = neighbor(n, dim) != MPI.PROC_NULL
 any_array(fields::GGField...)          = any([is_array(A.A) for A in fields])
 any_cuarray(fields::GGField...)        = any([is_cuarray(A.A) for A in fields])
 any_rocarray(fields::GGField...)       = any([is_rocarray(A.A) for A in fields])
+all_arrays(fields::GGField...)         = all([is_array(A.A) for A in fields])
+all_cuarrays(fields::GGField...)       = all([is_cuarray(A.A) for A in fields])
+all_rocarrays(fields::GGField...)      = all([is_rocarray(A.A) for A in fields])
 is_array(A::GGArray)                   = typeof(A) <: Array