Another mul! overload (#64)

* Another mul! overload

* Update ci.yml

* Update test_layoutarray.jl

Author: dlfivefifty

.github/workflows/ci.yml

@@ -11,7 +11,7 @@ jobs:
       matrix:
         version:
           - '1.5'
-          - '^1.6.0-0'
+          - '1'
         os:
           - ubuntu-latest
           - macOS-latest

Project.toml

@@ -1,7 +1,7 @@
 name = "ArrayLayouts"
 uuid = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"
 authors = ["Sheehan Olver <[email protected]>"]
-version = "0.7"
+version = "0.7.1"
 
 
 [deps]

src/mul.jl

@@ -163,8 +163,10 @@ macro layoutmul(Typ)
 
         LinearAlgebra.mul!(dest::AbstractMatrix, A::$Typ, b::AbstractMatrix) =
             ArrayLayouts.mul!(dest,A,b)
-        LinearAlgebra.mul!(dest::AbstractMatrix, A::$Typ, b::$Typ) =
-            ArrayLayouts.mul!(dest,A,b)
+        LinearAlgebra.mul!(dest::AbstractMatrix, A::AbstractMatrix, B::$Typ) =
+            ArrayLayouts.mul!(dest,A,B)
+        LinearAlgebra.mul!(dest::AbstractMatrix, A::$Typ, B::$Typ) =
+            ArrayLayouts.mul!(dest,A,B)
 
         Base.:*(A::$Typ, B::$Typ) = ArrayLayouts.mul(A,B)
         Base.:*(A::$Typ, B::AbstractMatrix) = ArrayLayouts.mul(A,B)

test/test_layoutarray.jl

@@ -106,6 +106,12 @@ MemoryLayout(::Type{MyVector}) = DenseColumnMajor()
             muladd!(1.0, Ones(5,5), A, 2.0, B)
         end
 
+        @testset "mul!" begin
+            @test mul!(B, A, A) ≈ A*A
+            @test mul!(B, A', A) ≈ A'*A
+            @test mul!(B, A, A') ≈ A*A'
+        end
+
         @testset "generic_blasmul!" begin
             A = BigFloat.(randn(5,5))
             Bin = BigFloat.(randn(5,5))