Add mapreduce_cuda for DataF

CliMA · Dec 5, 2023 · b6b84b7 · b6b84b7
1 parent 765e400
commit b6b84b7
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/Fields/mapreduce_cuda.jl b/src/Fields/mapreduce_cuda.jl
@@ -83,6 +83,21 @@ function LinearAlgebra.norm(
     end
 end
 
+function mapreduce_cuda(
+    f,
+    op,
+    field::Field{V};
+    weighting = false,
+    opargs...,
+) where {
+    S,
+    V <: DataLayouts.DataF{S},
+}
+    data = Fields.field_values(field)
+    pdata = parent(data)
+    return DataLayouts.DataF{S}(Array(Array(f(pdata))[1, :]))
+end
+
 function mapreduce_cuda(
     f,
     op,