From 044a43f42b6170488d0f3ab509a2a4d564eb8a0c Mon Sep 17 00:00:00 2001
From: Max Bernstein <tekknolagi@gmail.com>
Date: Fri, 10 Apr 2026 17:09:30 -0400
Subject: ZJIT: Trace infer_types as a sub-pass of other passes (#16714)

This helps us see how much time it takes in compiler tracing.
---
 zjit/src/hir.rs | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/zjit/src/hir.rs b/zjit/src/hir.rs
index 10c2f461e3..786a994d8b 100644
--- a/zjit/src/hir.rs
+++ b/zjit/src/hir.rs
@@ -4328,7 +4328,7 @@ impl Function {
                 }
             }
         }
-        self.infer_types();
+        crate::stats::trace_compile_phase("infer_types", || self.infer_types());
     }
 
     fn inline(&mut self) {
@@ -4382,7 +4382,7 @@ impl Function {
                 }
             }
         }
-        self.infer_types();
+        crate::stats::trace_compile_phase("infer_types", || self.infer_types());
     }
 
     fn load_shape(&mut self, block: BlockId, recv: InsnId) -> InsnId {
@@ -4670,7 +4670,7 @@ impl Function {
                 }
             }
         }
-        self.infer_types();
+        crate::stats::trace_compile_phase("infer_types", || self.infer_types());
     }
 
     fn gen_patch_points_for_optimized_ccall(&mut self, block: BlockId, recv_class: VALUE, method_id: ID, cme: *const rb_callable_method_entry_struct, state: InsnId) {
@@ -4989,7 +4989,7 @@ impl Function {
                 self.push_insn_id(block, insn_id);
             }
         }
-        self.infer_types();
+        crate::stats::trace_compile_phase("infer_types", || self.infer_types());
     }
 
     /// Convert `Send` instructions with no profile data into `SideExit` with recompile info.
@@ -5488,7 +5488,7 @@ impl Function {
             changed = true;
         }
         if changed {
-            self.infer_types();
+            crate::stats::trace_compile_phase("infer_types", || self.infer_types());
         }
     }
 
-- 
cgit v1.2.3