feat: Into<OpType> for custom ops (#731)

CQCL · Dec 1, 2023 · a69246b · a69246b
1 parent 39599d8
commit a69246b
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/ops/custom.rs b/src/ops/custom.rs
@@ -59,6 +59,13 @@ impl From<ExternalOp> for LeafOp {
     }
 }
 
+impl From<ExternalOp> for OpType {
+    fn from(value: ExternalOp) -> Self {
+        let leaf: LeafOp = value.into();
+        leaf.into()
+    }
+}
+
 impl ExternalOp {
     /// Name of the ExternalOp
     pub fn name(&self) -> SmolStr {
@@ -235,6 +242,19 @@ impl OpaqueOp {
     }
 }
 
+impl From<OpaqueOp> for LeafOp {
+    fn from(value: OpaqueOp) -> Self {
+        LeafOp::CustomOp(Box::new(ExternalOp::Opaque(value)))
+    }
+}
+
+impl From<OpaqueOp> for OpType {
+    fn from(value: OpaqueOp) -> Self {
+        let leaf: LeafOp = value.into();
+        leaf.into()
+    }
+}
+
 /// Resolve serialized names of operations into concrete implementation (OpDefs) where possible
 #[allow(dead_code)]
 pub fn resolve_extension_ops(