add invoke support

Author: johnzl-777

src/bloqade/gemini/analysis/measurement_validation/impls.py

@@ -1,5 +1,6 @@
 from kirin import ir, interp as _interp
 from kirin.analysis import ForwardFrame
+from kirin.dialects import func
 
 from bloqade import qubit, gemini
 from bloqade.analysis.address.lattice import AddressReg, AddressQubit
@@ -108,3 +109,21 @@ def terminal_measure(
             return (interp.lattice.bottom(),)
 
         return (interp.lattice.bottom(),)
+
+
+@func.dialect.register(key="gemini.validate.terminal_measurement")
+class Func(_interp.MethodTable):
+    @_interp.impl(func.Invoke)
+    def return_(
+        self,
+        interp: _GeminiTerminalMeasurementValidationAnalysis,
+        frame: ForwardFrame,
+        stmt: func.Invoke,
+    ):
+        _, ret = interp.call(
+            stmt.callee.code,
+            interp.method_self(stmt.callee),
+            *frame.get_values(stmt.inputs),
+        )
+
+        return (ret,)

test/gemini/test_logical_validation.py

@@ -137,6 +137,23 @@ def not_all_qubits_consumed():
     with pytest.raises(ValidationErrorGroup):
         validation_result.raise_if_invalid()
 
+    @gemini.logical.kernel
+    def terminal_measure_kernel(q):
+        return gemini.logical.terminal_measure(q)
+
+    @gemini.logical.kernel(no_raise=False, aggressive_unroll=True, typeinfer=True)
+    def terminal_measure_in_kernel():
+        q = squin.qalloc(10)
+        sub_qs = q[:2]
+        m = terminal_measure_kernel(sub_qs)
+        return m
+
+    validator = ValidationSuite([GeminiTerminalMeasurementValidation])
+    validation_result = validator.validate(terminal_measure_in_kernel)
+
+    with pytest.raises(ValidationErrorGroup):
+        validation_result.raise_if_invalid()
+
 
 def test_multiple_errors():
     did_error = False