Simplify map dispatch.

Author: maleadt

src/host/base.jl

@@ -1,24 +1,5 @@
 # common Base functionality
 
-allequal(x) = true
-allequal(x, y, z...) = x == y && allequal(y, z...)
-function Base.map!(f, y::AbstractGPUArray, xs::AbstractGPUArray...)
-    @assert allequal(size.((y, xs...))...)
-    return y .= f.(xs...)
-end
-function Base.map(f, y::AbstractGPUArray, xs::AbstractGPUArray...)
-    @assert allequal(size.((y, xs...))...)
-    return f.(y, xs...)
-end
-
-# Break ambiguities with base
-Base.map!(f, y::AbstractGPUArray) =
-    invoke(map!, Tuple{Any,AbstractGPUArray,Vararg{AbstractGPUArray}}, f, y)
-Base.map!(f, y::AbstractGPUArray, x::AbstractGPUArray) =
-    invoke(map!, Tuple{Any,AbstractGPUArray, Vararg{AbstractGPUArray}}, f, y, x)
-Base.map!(f, y::AbstractGPUArray, x1::AbstractGPUArray, x2::AbstractGPUArray) =
-    invoke(map!, Tuple{Any,AbstractGPUArray, Vararg{AbstractGPUArray}}, f, y, x1, x2)
-
 function Base.repeat(a::AbstractGPUVecOrMat, m::Int, n::Int = 1)
     o, p = size(a, 1), size(a, 2)
     b = similar(a, o*m, p*n)

src/host/broadcast.jl

@@ -82,3 +82,19 @@ end
 # `fill!` in general for all `GPUDestArray` so we just go straight to the fallback
 @inline Base.copyto!(dest::GPUDestArray, bc::Broadcasted{<:Broadcast.AbstractArrayStyle{0}}) =
     copyto!(dest, convert(Broadcasted{Nothing}, bc))
+
+
+## map
+
+allequal(x) = true
+allequal(x, y, z...) = x == y && allequal(y, z...)
+
+function Base.map!(f, y::AbstractGPUArray, xs::AbstractArray...)
+    @assert allequal(size.((y, xs...))...)
+    return y .= f.(xs...)
+end
+
+function Base.map(f, y::AbstractGPUArray, xs::AbstractArray...)
+    @assert allequal(size.((y, xs...))...)
+    return f.(y, xs...)
+end