Generalize interpreter traversal over CFG, Block, DiGraph, and UnGraph bodies

Cargo.toml

@@ -115,10 +115,14 @@ kirin-bitwise = { workspace = true }
 kirin-cf = { workspace = true }
 kirin-cmp = { workspace = true }
 kirin-constant = { workspace = true }
+kirin-constprop = { workspace = true }
 kirin-function = { workspace = true }
 kirin-scf = { workspace = true }
+kirin-interpreter = { workspace = true, features = ["derive"] }
 kirin-test-languages = { workspace = true, features = [
   "arith-function-language",
+  "graph-function-language",
+  "interpreter",
   "bitwise-function-language",
   "callable-language",
   "namespaced-language",

crates/kirin-constprop/src/context.rs

@@ -21,7 +21,9 @@
 
 use std::collections::{HashMap, HashSet};
 
-use kirin_interpreter::{CallContext, ContextInsensitive, InterpreterError, WideningStrategy};
+use kirin_interpreter::{
+    CallContext, ContextInsensitive, FunctionTarget, InterpreterError, WideningStrategy,
+};
 use kirin_ir::{CompileStage, Product, SpecializedFunction};
 
 use crate::ConstPropValue;
@@ -67,12 +69,9 @@ impl Default for ConstPropContext {
 impl CallContext<ConstPropValue> for ConstPropContext {
     type Key = (CompileStage, SpecializedFunction, CallCtx);
 
-    fn key(
-        &mut self,
-        stage: CompileStage,
-        function: SpecializedFunction,
-        args: &Product<ConstPropValue>,
-    ) -> Self::Key {
+    fn key(&mut self, target: &FunctionTarget, args: &Product<ConstPropValue>) -> Self::Key {
+        let stage = target.stage;
+        let function = target.function;
         let ctx = match all_const(args) {
             Some(consts) => {
                 let admitted = self.admitted.entry((stage, function)).or_default();