[new] Infer extension requirements (#330)

Add a module `extension::infer`, which given a Hugr attempts to infer the resource requirements where they are left open. Addresses #425 --------- Co-authored-by: Seyon Sivarajah <[email protected]>
CQCL · Aug 25, 2023 · 95c0b56 · 95c0b56
1 parent 2013622
commit 95c0b56
Show file tree

Hide file tree

Showing 6 changed files with 1,057 additions and 36 deletions.
diff --git a/src/builder/dataflow.rs b/src/builder/dataflow.rs
@@ -96,8 +96,9 @@ impl DFGBuilder<Hugr> {
 }
 
 impl HugrBuilder for DFGBuilder<Hugr> {
-    fn finish_hugr(self) -> Result<Hugr, ValidationError> {
-        self.base.validate()?;
+    fn finish_hugr(mut self) -> Result<Hugr, ValidationError> {
+        let closure = self.base.infer_extensions()?;
+        self.base.validate_with_extension_closure(closure)?;
         Ok(self.base)
     }
 }

diff --git a/src/extension.rs b/src/extension.rs
@@ -17,6 +17,9 @@ use crate::types::type_param::{check_type_arg, TypeArgError};
 use crate::types::type_param::{TypeArg, TypeParam};
 use crate::types::CustomType;
 
+mod infer;
+pub use infer::{infer_extensions, ExtensionSolution, InferExtensionError};
+
 mod op_def;
 pub use op_def::{CustomSignatureFunc, OpDef};
 mod type_def;