From f3df04d8f88cdf2466b886d117e8d28ca537b4a1 Mon Sep 17 00:00:00 2001
From: Pratyai Mazumder <pratyai.mazumder@gmail.com>
Date: Sun, 9 Jun 2024 21:00:17 +0200
Subject: [PATCH] Count another location of `KKT.solve!()` call in the final
 stats (#147)

(previously unreported).
---
 src/IPM/HSD/step.jl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/IPM/HSD/step.jl b/src/IPM/HSD/step.jl
index cc7b7abc..98618c04 100644
--- a/src/IPM/HSD/step.jl
+++ b/src/IPM/HSD/step.jl
@@ -59,7 +59,9 @@ function compute_step!(hsd::HSD{T, Tv}, params::IPMOptions{T}) where{T, Tv<:Abst
     hx = tzeros(Tv, n)
     hy = tzeros(Tv, m)
     ξ_ = @. (dat.c - ((pt.zl / pt.xl) * dat.l) * dat.lflag - ((pt.zu / pt.xu) * dat.u) * dat.uflag)
-    KKT.solve!(hx, hy, hsd.kkt, dat.b, ξ_)
+    @timeit hsd.timer "Newton" begin
+      @timeit hsd.timer "KKT" KKT.solve!(hx, hy, hsd.kkt, dat.b, ξ_)
+    end
 
     # Recover h0 = ρg + κ / τ - c'hx + b'hy - u'hz
     # Some of the summands may take large values,