feat: Add missing tket2 quantum gates, generalise RotationCompile

guppylang/prelude/_internal/compiler/quantum.py

@@ -70,7 +70,7 @@ def __init__(self, opname: str):
     def compile_with_inouts(self, args: list[Wire]) -> CallReturnWires:
         from guppylang.prelude._internal.util import quantum_op
 
-        [q, angle] = args
+        qs, angle = args[:-1], args[-1]
         [halfturns] = self.builder.add_op(ops.UnpackTuple([FLOAT_T]), angle)
         [mb_rotation] = self.builder.add_op(from_halfturns(), halfturns)
 
@@ -81,11 +81,11 @@ def compile_with_inouts(self, args: list[Wire]) -> CallReturnWires:
         with conditional.add_case(1) as case:
             case.set_outputs(*case.inputs())
 
-        q = self.builder.add_op(
+        qs = self.builder.add_op(
             quantum_op(self.opname)(
-                ht.FunctionType([ht.Qubit, ROTATION_T], [ht.Qubit]), []
+                ht.FunctionType([ht.Qubit for _ in qs] + [ROTATION_T], [ht.Qubit for _ in qs]), []
             ),
-            q,
+            *qs,
             conditional,
         )
-        return CallReturnWires(regular_returns=[], inout_returns=[q])
+        return CallReturnWires(regular_returns=[], inout_returns=qs)

guppylang/prelude/quantum.py

@@ -9,6 +9,7 @@
 from guppylang.decorator import guppy
 from guppylang.prelude._internal.compiler.quantum import (
     HSERIES_EXTENSION,
+    QUANTUM_EXTENSION,
     MeasureReturnCompiler,
     RotationCompiler,
 )
@@ -35,6 +36,10 @@ def h(q: qubit) -> None: ...
 def cz(control: qubit, target: qubit) -> None: ...
 
 
+@guppy.hugr_op(quantum_op("CY"))
+def cy(control: qubit, target: qubit) -> None: ...
+
+
 @guppy.hugr_op(quantum_op("CX"))
 def cx(control: qubit, target: qubit) -> None: ...
 
@@ -79,6 +84,18 @@ def rz(q: qubit, angle: angle) -> None: ...
 def rx(q: qubit, angle: angle) -> None: ...
 
 
+@guppy.custom(RotationCompiler("Ry"))
+def ry(q: qubit, angle: angle) -> None: ...
+
+
+@guppy.custom(RotationCompiler("CRz"))
+def crz(control: qubit, target: qubit, angle: angle) -> None: ...
+
+
+@guppy.hugr_op(quantum_op("Toffoli"))
+def toffoli(control1: qubit, control2: qubit, target: qubit) -> None: ...
+
+
 @guppy.hugr_op(quantum_op("QAlloc"))
 def dirty_qubit() -> qubit: ...