More diag mul methods

Author: kshyatt

src/host/linalg.jl

@@ -259,8 +259,8 @@ function LinearAlgebra.mul!(C::Diagonal{<:Any, <:AbstractGPUArray},
 end
 
 function LinearAlgebra.mul!(C::Diagonal{<:Any, <:AbstractGPUArray},
-                            A::AbstractGPUArray,
-                            B::AbstractGPUArray)
+                            A::Union{AbstractGPUArray, Adjoint{T,<:AbstractGPUArray{T}}, Transpose{T,<:AbstractGPUArray{T}}},
+                            B::Union{AbstractGPUArray, Adjoint{T,<:AbstractGPUArray{T}}, Transpose{T,<:AbstractGPUArray{T}}}) where {T}
     dc = C.diag
     d  = length(dc)
     m, n   = size(A, 1), size(A, 2)

test/testsuite/linalg.jl

@@ -255,6 +255,11 @@
                 C = Diagonal(d)
                 mul!(C, a, b)
                 @test collect(C) ≈ Diagonal(collect(a) * collect(b)) 
+                a = transpose(AT(diagm(rand(elty, n))))
+                b = adjoint(AT(diagm(rand(elty, n))))
+                C = Diagonal(d)
+                mul!(C, a, b)
+                @test collect(C) ≈ Diagonal(collect(a) * collect(b)) 
             end
         end