support adjoint(::Diagonal{<:CuVector{<:Complex}}) (e.g. mul! resulted in scalar-indexing) + test

Author: tam724

lib/cublas/linalg.jl

@@ -339,6 +339,8 @@ function LinearAlgebra.inv(D::Diagonal{T, <:CuArray{T}}) where {T}
     Diagonal(Di)
 end
 
+LinearAlgebra.adjoint(D::Diagonal{T, <:CuVector{T}}) where T <: Complex = Diagonal(map(adjoint, D.diag))
+
 LinearAlgebra.rdiv!(A::CuArray, D::Diagonal) =  _rdiv!(A, A, D)
 
 Base.:/(A::CuArray, D::Diagonal) = _rdiv!(similar(A, typeof(oneunit(eltype(A)) / oneunit(eltype(D)))), A, D)

test/libraries/cublas/extensions.jl

@@ -538,11 +538,23 @@ k = 13
             mul!(d_XA, d_X, d_A)
             Array(d_XA) ≈ Diagonal(x) * A
 
+            XA = rand(elty,m,n)
+            d_XA = CuArray(XA)
+            d_X = Diagonal(d_x)
+            mul!(d_XA, d_X', d_A)
+            Array(d_XA) ≈ Diagonal(x)' * A
+
             AY = rand(elty,m,n)
             d_AY = CuArray(AY)
             d_Y = Diagonal(d_y)
             mul!(d_AY, d_A, d_Y)
             Array(d_AY) ≈ A * Diagonal(y)
+            
+            AY = rand(elty,m,n)
+            d_AY = CuArray(AY)
+            d_Y = Diagonal(d_y)
+            mul!(d_AY, d_A, d_Y')
+            Array(d_AY) ≈ A * Diagonal(y)'
 
             YA = rand(elty,n,m)
             d_YA = CuArray(YA)