Merge pull request #584 from judober/Increase_Matrix_Div_Performance

Increase Performance of \ (and /) for small square matrix

Author: c42f

src/solve.jl

@@ -26,6 +26,22 @@ end
             (a[1,1]*a[2,2] - a[1,2]*a[2,1])*b[3]) / d )
 end
 
+for Sa in [(2,2), (3,3)]  # not needed for Sa = (1, 1);
+    @eval begin
+        @inline function solve(::Size{$Sa}, ::Size{Sb}, a::StaticMatrix{<:Any, <:Any, Ta}, b::StaticMatrix{<:Any, <:Any, Tb}) where {Sb, Ta, Tb}
+            d = det(a)
+            T = typeof((one(Ta)*zero(Tb) + one(Ta)*zero(Tb))/d)
+            c = similar(b, T)
+            for col = 1:Sb[2]
+                @inbounds c[:, col] = solve(Size($Sa), Size($Sa[1],), a, b[:, col])
+            end
+            return similar_type(b, T)(c)
+        end
+    end # @eval
+end
+
+
+
 @generated function solve(::Size{Sa}, ::Size{Sb}, a::StaticMatrix{<:Any, <:Any, Ta}, b::StaticVecOrMat{Tb}) where {Sa, Sb, Ta, Tb}
     if Sa[end] != Sb[1]
         throw(DimensionMismatch("right hand side B needs first dimension of size $(Sa[end]), has size $Sb"))