tweak in cholldiv

droodman · Dec 24, 2023 · e530e1a · e530e1a
1 parent 49cf7a5
commit e530e1a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/utilities.jl b/src/utilities.jl
@@ -350,11 +350,12 @@ function cholldiv!(ch::CholeskyPivoted{T}, Y::AbstractVecOrMat{T}) where T  # ad
 	  if rnk == len
 	    ldiv!(ch, Y)
 	  else
+			invpiv = invperm(ch.piv)
 	    for v ∈ eachcol(Y)
 	      permute!(v, ch.piv)
 	      v[rnk+1:len] .= zero(T)
 	      LAPACK.potrs!(ch.uplo, view(ch.factors, 1:rnk, 1:rnk), view(v, 1:rnk, :))
-	      invpermute!(v, ch.piv)
+	      permute!(v, invpiv)
 	    end
 	  end
 	end