Preserve 0-dim arrays in map (#599)

Co-authored-by: Tim Besard <[email protected]>

Author: mtfishman

Project.toml

@@ -1,6 +1,6 @@
 name = "GPUArrays"
 uuid = "0c68f7d7-f131-5f86-a1c3-88cf8149b2d7"
-version = "11.2.2"
+version = "11.2.3"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/host/broadcast.jl

@@ -82,10 +82,11 @@ end
 allequal(x) = true
 allequal(x, y, z...) = x == y && allequal(y, z...)
 
-function Base.map(f, xs::AnyGPUArray...)
+function Base.map(f, x1::AnyGPUArray, xrest::AnyGPUArray...)
+    xs = (x1, xrest...)
     # if argument sizes match, their shape needs to be preserved
     if allequal(size.(xs)...)
-         return f.(xs...)
+        return Broadcast.broadcast_preserving_zero_d(f, xs...)
     end
 
     # if not, treat them as iterators

test/testsuite/broadcasting.jl

@@ -158,6 +158,11 @@ function broadcasting(AT, eltypes)
             @test compare(AT, rand(ET, 2,2), rand(ET, 2)) do x,y
                 map(+, x, y)
             end
+            ############
+            # issue #598
+            @test compare(AT, rand(ET, ()), rand(ET, ())) do x, y
+                map(+, x, y)
+            end
         end
     end