Make type_of use union-find

tekknolagi · tekknolagi · commit 65efef5c563e · 2025-04-23T16:24:30.000-04:00
Otherwise we might have stale types floating around
diff --git a/zjit/src/hir.rs b/zjit/src/hir.rs
@@ -789,7 +789,7 @@ impl Function {
 
     fn type_of(&self, insn: InsnId) -> Type {
         assert!(self.insns[insn.0].has_output());
-        self.insn_types[insn.0]
+        self.insn_types[self.union_find.find_const(insn).0]
     }
 
     /// Check if the type of `insn` is a subtype of `ty`.

Original file line number	Diff line number	Diff line change
`@@ -789,7 +789,7 @@ impl Function {`
`789`	`789`
`790`	`790`	`fn type_of(&self, insn: InsnId) -> Type {`
`791`	`791`	`assert!(self.insns[insn.0].has_output());`
`792`		`- self.insn_types[insn.0]`
	`792`	`+ self.insn_types[self.union_find.find_const(insn).0]`
`793`	`793`	`}`
`794`	`794`
`795`	`795`	/// Check if the type of `insn` is a subtype of `ty`.