Update elemwise.jl

Author: arhik

src/ops/elemwise.jl

@@ -8,7 +8,7 @@ function broadcast_kernel(op::Function, x::WgpuArray{T, N}, y::WgpuArray{T, N},
 	out[gId] = op(x[gId], y[gId])
 end
 
-function broadcast_kernel(op::Function, x::WgpuArray{T, N}, y::Number, out::WgpuArray{T, N}) where {T, N}
+function broadcast_kernel(op::Function, x::WgpuArray{T, N}, y::Float32, out::WgpuArray{T, N}) where {T, N}
 	xdim = workgroupDims.x
 	ydim = workgroupDims.y
 	gIdx = workgroupId.x*xdim + localId.x
@@ -44,15 +44,15 @@ function elwise(f::Function, a::Number, b::WgpuArray{T, N}) where {T, N}
 	return out
 end
 
-# Base.:+(a::WgpuArray{T, N}, b::WgpuArray{T, N}) where {T, N} = elwise(+, a, b)
-# Base.:-(a::WgpuArray{T, N}, b::WgpuArray{T, N}) where {T, N} = elwise(-, a, b)
-# Base.:*(a::WgpuArray{T, N}, b::WgpuArray{T, N}) where {T, N} = elwise(*, a, b)
-# Base.:/(a::WgpuArray{T, N}, b::WgpuArray{T, N}) where {T, N} = elwise(/, a, b)
-# Base.:+(a::WgpuArray{T, N}, b::Number) where {T, N} = elwise(+, a, b)
-# Base.:-(a::WgpuArray{T, N}, b::Number) where {T, N} = elwise(-, a, b)
-# Base.:*(a::WgpuArray{T, N}, b::Number) where {T, N} = elwise(*, a, b)
-# Base.:/(a::WgpuArray{T, N}, b::Number) where {T, N} = elwise(/, a, b)
-# Base.:+(a::Number, b::WgpuArray{T, N}) where {T, N} = elwise(+, a, b)
-# Base.:-(a::Number, b::WgpuArray{T, N}) where {T, N} = elwise(-, a, b)
-# Base.:*(a::Number, b::WgpuArray{T, N}) where {T, N} = elwise(*, a, b)
-# Base.:/(a::Number, b::WgpuArray{T, N}) where {T, N} = elwise(/, a, b)
+Base.:+(a::WgpuArray{T, N}, b::WgpuArray{T, N}) where {T, N} = elwise(+, a, b)
+Base.:-(a::WgpuArray{T, N}, b::WgpuArray{T, N}) where {T, N} = elwise(-, a, b)
+Base.:*(a::WgpuArray{T, N}, b::WgpuArray{T, N}) where {T, N} = elwise(*, a, b)
+Base.:/(a::WgpuArray{T, N}, b::WgpuArray{T, N}) where {T, N} = elwise(/, a, b)
+Base.:+(a::WgpuArray{T, N}, b::Number) where {T, N} = elwise(+, a, b)
+Base.:-(a::WgpuArray{T, N}, b::Number) where {T, N} = elwise(-, a, b)
+Base.:*(a::WgpuArray{T, N}, b::Number) where {T, N} = elwise(*, a, b)
+Base.:/(a::WgpuArray{T, N}, b::Number) where {T, N} = elwise(/, a, b)
+Base.:+(a::Number, b::WgpuArray{T, N}) where {T, N} = elwise(+, a, b)
+Base.:-(a::Number, b::WgpuArray{T, N}) where {T, N} = elwise(-, a, b)
+Base.:*(a::Number, b::WgpuArray{T, N}) where {T, N} = elwise(*, a, b)
+Base.:/(a::Number, b::WgpuArray{T, N}) where {T, N} = elwise(/, a, b)