fix test, interface

Author: oscardssmith

src/butterflylu.jl

@@ -24,9 +24,10 @@ struct 🦋workspace{T}
     U::Matrix{T}
     V::Matrix{T}
     out::Vector{T}
+    n::Int
     function 🦋workspace(A, b, ::Val{SEED} = Val(888)) where {SEED}
-        M = size(A, 1)
-        out = similar(b, M)
+        N = size(A, 1)
+        out = similar(b, N)
         if (M % 4 != 0)
             A = pad!(A)
             xn = 4 - M % 4
@@ -35,20 +36,20 @@ struct 🦋workspace{T}
         U, V = (similar(A), similar(A))
         ws = 🦋generate_random!(A)
         materializeUV(U, V, ws)
-        new{eltype(A)}(A, b, ws, U, V, out)
+        new{eltype(A)}(A, b, ws, U, V, out, N)
     end
 end
 
-function 🦋solve!(workspace::🦋workspace, M, thread)
-    (;A, b, ws, U, V, out) = workspace
+function 🦋solve!(workspace::🦋workspace, thread)
+    (;A, b, ws, U, V, out, 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)
-    out .= @view b[1:M]  
+    out .= @view b[1:N]
     out
 end
 

test/runtests.jl

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