More diag mul methods

[kshyatt]

No description provided.

[github-actions]

Your PR requires formatting changes to meet the project's style guidelines.
Please consider running Runic (git runic master) to apply these changes.

Click here to view the suggested changes.

diff --git a/src/host/linalg.jl b/src/host/linalg.jl
index 0002fdd..cccc004 100644
--- a/src/host/linalg.jl
+++ b/src/host/linalg.jl
@@ -266,8 +266,9 @@ function LinearAlgebra.mul!(C::Diagonal{<:Any, <:AbstractGPUArray},
 end
 
 function LinearAlgebra.mul!(C::Diagonal{<:Any, <: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}
+        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)
@@ -282,7 +283,8 @@ end
 
 function LinearAlgebra.mul!(B::AbstractGPUVecOrMat,
                             D::Diagonal{<:Any, <:AbstractGPUArray},
-                            A::Union{AbstractGPUArray, Adjoint{T,<:AbstractGPUArray{T}}, Transpose{T,<:AbstractGPUArray{T}}}) where {T}
+        A::Union{AbstractGPUArray, Adjoint{T, <:AbstractGPUArray{T}}, Transpose{T, <:AbstractGPUArray{T}}}
+    ) where {T}
     dd = D.diag
     d = length(dd)
     m, n = size(A, 1), size(A, 2)
@@ -295,9 +297,10 @@ end
 
 function LinearAlgebra.mul!(B::AbstractGPUVecOrMat,
                             D::Diagonal{<:Any, <:AbstractGPUArray},
-                            A::Union{AbstractGPUArray, Adjoint{T,<:AbstractGPUArray{T}}, Transpose{T,<:AbstractGPUArray{T}}},
+        A::Union{AbstractGPUArray, Adjoint{T, <:AbstractGPUArray{T}}, Transpose{T, <:AbstractGPUArray{T}}},
                             α::Number,
-                            β::Number) where {T}
+        β::Number
+    ) where {T}
     dd = D.diag
     d = length(dd)
     m, n = size(A, 1), size(A, 2)
diff --git a/test/testsuite/linalg.jl b/test/testsuite/linalg.jl
index d77a2bb..484ab30 100644
--- a/test/testsuite/linalg.jl
+++ b/test/testsuite/linalg.jl
@@ -266,7 +266,7 @@
                 b = adjoint(AT(diagm(rand(elty, n))))
                 C = Diagonal(d)
                 mul!(C, a, b)
-                @test collect(C) ≈ Diagonal(collect(a) * collect(b)) 
+                @test collect(C) ≈ Diagonal(collect(a) * collect(b))
             end
         end