more matrix types for multiplication

Author: mateuszbaran

src/matrix_multiply.jl

@@ -15,6 +15,9 @@ const StaticMatMulLike{s1, s2, T} = Union{
     Hermitian{T, <:StaticMatrix{s1, s2, T}},
     LowerTriangular{T, <:StaticMatrix{s1, s2, T}},
     UpperTriangular{T, <:StaticMatrix{s1, s2, T}},
+    UnitLowerTriangular{T, <:StaticMatrix{s1, s2, T}},
+    UnitUpperTriangular{T, <:StaticMatrix{s1, s2, T}},
+    UpperHessenberg{T, <:StaticMatrix{s1, s2, T}},
     Adjoint{T, <:StaticMatrix{s1, s2, T}},
     Transpose{T, <:StaticMatrix{s1, s2, T}}}
 
@@ -67,6 +70,15 @@ end
 function gen_by_access(expr_gen, a::Type{<:LowerTriangular{<:Any, <:StaticMatrix}}, asym = :a)
     return expr_gen(:lower_triangular)
 end
+function gen_by_access(expr_gen, a::Type{<:UnitUpperTriangular{<:Any, <:StaticMatrix}}, asym = :a)
+    return expr_gen(:unit_upper_triangular)
+end
+function gen_by_access(expr_gen, a::Type{<:UnitLowerTriangular{<:Any, <:StaticMatrix}}, asym = :a)
+    return expr_gen(:unit_lower_triangular)
+end
+function gen_by_access(expr_gen, a::Type{<:UpperHessenberg{<:Any, <:StaticMatrix}}, asym = :a)
+    return expr_gen(:upper_hessenberg)
+end
 function gen_by_access(expr_gen, a::Type{<:Transpose{<:Any, <:StaticVecOrMat}}, asym = :a)
     return expr_gen(:transpose)
 end
@@ -127,6 +139,27 @@ function gen_by_access(expr_gen, a::Type{<:LowerTriangular{<:Any, <:StaticMatrix
         end)
     end
 end
+function gen_by_access(expr_gen, a::Type{<:UnitUpperTriangular{<:Any, <:StaticMatrix}}, b::Type)
+    return quote
+        return $(gen_by_access(b, :b) do access_b
+            expr_gen(:unit_upper_triangular, access_b)
+        end)
+    end
+end
+function gen_by_access(expr_gen, a::Type{<:UnitLowerTriangular{<:Any, <:StaticMatrix}}, b::Type)
+    return quote
+        return $(gen_by_access(b, :b) do access_b
+            expr_gen(:unit_lower_triangular, access_b)
+        end)
+    end
+end
+function gen_by_access(expr_gen, a::Type{<:UpperHessenberg{<:Any, <:StaticMatrix}}, b::Type)
+    return quote
+        return $(gen_by_access(b, :b) do access_b
+            expr_gen(:upper_hessenberg, access_b)
+        end)
+    end
+end
 function gen_by_access(expr_gen, a::Type{<:Transpose{<:Any, <:StaticMatrix}}, b::Type)
     return quote
         return $(gen_by_access(b, :b) do access_b
@@ -205,6 +238,28 @@ function uplo_access(sa, asym, k, j, uplo)
         else
             return :(zero(T))
         end
+    elseif uplo == :unit_upper_triangular
+        if k < j
+            return :($asym[$(LinearIndices(sa)[k, j])])
+        elseif k == j
+            return :(oneunit(T))
+        else
+            return :(zero(T))
+        end
+    elseif uplo == :unit_lower_triangular
+        if k > j
+            return :($asym[$(LinearIndices(sa)[k, j])])
+        elseif k == j 
+            return :(oneunit(T))
+        else
+            return :(zero(T))
+        end
+    elseif uplo == :upper_hessenberg
+        if k <= j+1
+            return :($asym[$(LinearIndices(sa)[k, j])])
+        else
+            return :(zero(T))
+        end
     elseif uplo == :transpose
         return :($asym[$(LinearIndices(reverse(sa))[j, k])])
     elseif uplo == :ajoint

test/matrix_multiply.jl

@@ -8,6 +8,9 @@ mul_wrappers = [
     m -> Hermitian(m, :L),
     m -> UpperTriangular(m),
     m -> LowerTriangular(m),
+    m -> UnitUpperTriangular(m),
+    m -> UnitLowerTriangular(m),
+    m -> UpperHessenberg(m),
     m -> adjoint(m),
     m -> transpose(m)]