Restrict check_mul_axes to specific Q-matrix-types (#103)

* Restrict `check_mul_axes` to specific Q-matrix-types

* Update Project.toml

* Update ArrayLayouts.jl

Author: dkarrasch

Project.toml

@@ -1,7 +1,7 @@
 name = "ArrayLayouts"
 uuid = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"
 authors = ["Sheehan Olver <[email protected]>"]
-version = "0.8.9"
+version = "0.8.10"
 
 [deps]
 FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"

src/ArrayLayouts.jl

@@ -34,7 +34,8 @@ import Base.Broadcast: BroadcastStyle, AbstractArrayStyle, Broadcasted, broadcas
                         combine_eltypes, DefaultArrayStyle, instantiate, materialize,
                         materialize!, eltypes
 
-import LinearAlgebra: AbstractTriangular, AbstractQ, checksquare, pinv, fill!, tilebufsize, factorize, qr, lu, cholesky,
+import LinearAlgebra: AbstractTriangular, AbstractQ, QRCompactWYQ, QRPackedQ, checksquare, pinv,
+                        fill!, tilebufsize, factorize, qr, lu, cholesky,
                         norm2, norm1, normInf, normMinusInf, qr, lu, qr!, lu!, AdjOrTrans, HermOrSym, AdjointAbsVec,
                         TransposeAbsVec, cholcopy, checknonsingular, _apply_ipiv_rows!, ipiv2perm, RealHermSymComplexHerm, chkfullrank
 

src/mul.jl

@@ -95,7 +95,7 @@ function check_mul_axes(A, B, C...)
 end
 
 # we need to special case AbstractQ as it allows non-compatiple multiplication
-function check_mul_axes(A::AbstractQ, B, C...)
+function check_mul_axes(A::Union{QRCompactWYQ,QRPackedQ}, B, C...)
     axes(A.factors, 1) == axes(B, 1) || axes(A.factors, 2) == axes(B, 1) ||
         throw(DimensionMismatch("First axis of B, $(axes(B,1)) must match either axes of A, $(axes(A))"))
     check_mul_axes(B, C...)