fix

Author: oscardssmith

src/butterflylu.jl

@@ -24,31 +24,32 @@ struct 🦋workspace{T}
     U::Matrix{T}
     V::Matrix{T}
     out::Vector{T}
+    tmp::Vector{T}
     n::Int
     function 🦋workspace(A, b, ::Val{SEED} = Val(888)) where {SEED}
         n = size(A, 1)
-        out = similar(b, n)
+        out = similar(b)
         if (n % 4 != 0)
             A = pad!(A)
             xn = 4 - n % 4
             b = [b; rand(xn)]
         end
+        tmp = similar(b)
         U, V = (similar(A), similar(A))
         ws = 🦋generate_random!(A)
         materializeUV(U, V, ws)
-        new{eltype(A)}(A, b, ws, U, V, out, n)
+        new{eltype(A)}(A, b, ws, U, V, out, tmp, n)
     end
 end
 
 function 🦋solve!(workspace::🦋workspace, thread)
-    (;A, b, ws, U, V, out, n) = workspace
+    (;A, b, ws, U, V, out, tmp, n) = workspace
     🦋mul!(A, ws)
     F = RecursiveFactorization.lu!(A, Val(false), thread)
 
-    mul!(b, U', b)
-    ldiv!(b, UnitLowerTriangular(F.factors), b, thread)
-    ldiv!(b, UpperTriangular(F.factors), b, thread)
-    mul!(b, V, b)
+    mul!(tmp, U', b)
+    ldiv!(F, tmp, thread)
+    mul!(b, V, tmp)
     out .= @view b[1:n]
     out
 end

test/runtests.jl

@@ -81,7 +81,7 @@ end
     for i in 790 : 810
         A = wilkinson(i)
         b = rand(i)
-        ws = RecursiveFactorization.🦋workspace(copy(A), copy(b))    
+        ws = RecursiveFactorization.🦋workspace(copy(A), copy(b))
         out = RecursiveFactorization.🦋solve!(ws, Val(true))
         @test norm(A * out .- b) <= 1e-10
     end