Solve 1x1 factorization problem

Closes #128
MichielStock · Jun 27, 2023 · 93b444c · 93b444c
1 parent a05cd80
commit 93b444c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/vectrick.jl b/src/vectrick.jl
@@ -165,7 +165,7 @@ for TC in [:AbstractVector, :AbstractMatrix],
         #       factorization up to the user/Julia
         factors = ntuple(length(matrices)) do i
             m = matrices[i]
-            return (m isa Factorization) ? m : factorize(m)
+            return (m isa Factorization) ? m : unscalar(factorize(m))
         end
 
         if length(factors) == 2
@@ -190,6 +190,10 @@ for TC in [:AbstractVector, :AbstractMatrix],
     end
 end
 
+# Utility function to handle factorize(one_by_one)::Number
+unscalar(n::Number) = [n;;]
+unscalar(x) = x
+
 function Base.:*(K::GeneralizedKroneckerProduct, v::AbstractVector)
     return mul!(Vector{promote_type(eltype(v), eltype(K))}(undef, first(size(K))), K, v)
 end