Merge branch 'main' into release-plz-2024-12-16T14-23-11Z

hugr-core/src/extension/op_def.rs

@@ -245,6 +245,9 @@ impl SignatureFunc {
             SignatureFunc::MissingValidateFunc(ts) => (ts, args),
         };
         let mut res = pf.instantiate(args)?;
+
+        // Automatically add the extensions where the operation is defined to
+        // the runtime requirements of the op.
         res.runtime_reqs.insert(def.extension.clone());
 
         // If there are any row variables left, this will fail with an error:

hugr-core/src/extension/resolution.rs

@@ -10,7 +10,7 @@
 //! required across the HUGR.
 //!
 //! This is distinct from _runtime_ extension requirements, which are defined
-//! more granularly in each function signature by the `required_extensions`
+//! more granularly in each function signature by the `runtime_reqs`
 //! field. See the `extension_inference` feature and related modules for that.
 //!
 //! Note: These procedures are only temporary until `hugr-model` is stabilized.

hugr-core/src/extension/resolution/ops.rs

@@ -93,7 +93,7 @@ pub(crate) fn resolve_op_extensions<'e>(
             cause: e,
         })?;
 
-    if opaque.signature() != ext_op.signature() {
+    if opaque.signature().io() != ext_op.signature().io() {
         return Err(OpaqueOpError::SignatureMismatch {
             node,
             extension: opaque.extension().clone(),

hugr-core/src/hugr.rs

@@ -197,7 +197,7 @@ impl Hugr {
     ///
     /// This is distinct from _runtime_ extension requirements computed in
     /// [`Hugr::infer_extensions`], which are computed more granularly in each
-    /// function signature by the `required_extensions` field and define the set
+    /// function signature by the `runtime_reqs` field and define the set
     /// of capabilities required by the runtime to execute each function.
     ///
     /// Updates the internal extension registry with the extensions used in the

hugr-py/src/hugr/ext.py

@@ -236,6 +236,13 @@ def instantiate(
             concrete_signature: Concrete function type of the operation, only required
             if the operation is polymorphic.
         """
+        # Add the extension where the operation is defined as a runtime requirement.
+        # We don't store this in the json definition as it is redundant information.
+        if concrete_signature is not None:
+            concrete_signature = concrete_signature.with_runtime_reqs(
+                [self.get_extension().name]
+            )
+
         return ops.ExtOp(self, concrete_signature, list(args or []))