From 4911d5bfbd51d585c6da8b2263d0720d87acf6d9 Mon Sep 17 00:00:00 2001 From: Andrei Zhabinski Date: Sat, 8 Feb 2020 21:34:07 +0300 Subject: [PATCH] Fix unbroadcast_prod_*() for CuArrays --- src/helpers.jl | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/helpers.jl b/src/helpers.jl index 4c08db4..70e0442 100644 --- a/src/helpers.jl +++ b/src/helpers.jl @@ -74,10 +74,10 @@ end unbroadcast_prod_y(x::AbstractArray, y::AbstractArray, Δ) = unbroadcast_prod_x(y, x, Δ) -unbroadcast_prod_x(x::Number, y::AbstractArray, Δ) = unbroadcast_prod_x([x], y, Δ)[1] -unbroadcast_prod_x(x::AbstractArray, y::Number, Δ) = unbroadcast_prod_x(x, [y], Δ) -unbroadcast_prod_y(x::AbstractArray, y::Number, Δ) = unbroadcast_prod_y(x, [y], Δ)[1] -unbroadcast_prod_y(x::Number, y::AbstractArray, Δ) = unbroadcast_prod_y([x], y, Δ) +unbroadcast_prod_x(x::Number, y::AbstractArray, Δ) = unbroadcast_prod_x(convert(typeof(y), [x]), y, Δ)[1] +unbroadcast_prod_x(x::AbstractArray, y::Number, Δ) = unbroadcast_prod_x(x, convert(typeof(x), [y]), Δ) +unbroadcast_prod_y(x::AbstractArray, y::Number, Δ) = unbroadcast_prod_y(x, convert(typeof(x), [y]), Δ)[1] +unbroadcast_prod_y(x::Number, y::AbstractArray, Δ) = unbroadcast_prod_y(convert(typeof(y), [x]), y, Δ) untranspose_vec(ds::Transpose{T, <:AbstractVector{T}}) where T = transpose(ds)