update

Author: josephleekl

frontend/catalyst/jax_primitives.py

@@ -391,7 +391,7 @@ def wrapper(*args, **kwargs):
 
 
 def decomposition_rule(
-    func=None, *, is_qreg=True, num_params=0, base: qml.operation.Operator = None
+    func=None, *, is_qreg=True, num_params=0, op: qml.operation.Operator = None
 ):
     """
     Denotes the creation of a quantum definition in the intermediate representation.
@@ -403,13 +403,13 @@ def decomposition_rule(
 
     if func is None:
         return functools.partial(
-            decomposition_rule, is_qreg=is_qreg, num_params=num_params, base=base
+            decomposition_rule, is_qreg=is_qreg, num_params=num_params, op=op
         )
 
     @functools.wraps(func)
     def wrapper(*args, **kwargs):
-        if base is not None:
-            new_func = functools.partial(func, **base.hyperparameters)
+        if op is not None:
+            new_func = functools.partial(func,wires=op.wires, **op.hyperparameters)
             jaxpr = jax.make_jaxpr(new_func)(*args, **kwargs)
         else:
             jaxpr = jax.make_jaxpr(func)(*args, **kwargs)
@@ -418,7 +418,7 @@ def wrapper(*args, **kwargs):
             func_jaxpr=jaxpr,
             is_qreg=is_qreg,
             num_params=num_params,
-            op_name=base.name if base else "",
+            op_name=op.name if op else "",
         )
 
     return wrapper

test_new_decomp_adjoint.py

@@ -15,11 +15,35 @@
 qml.decomposition.enable_graph()
 
 
-@decomposition_rule(base=qml.adjoint(qml.RZ(float, wires=0)))
+@decomposition_rule(op=qml.adjoint(qml.RZ(float, wires=1)))
 def adjoint_rotation(phi, wires, base, **__):
     """Decompose the adjoint of a rotation operator by inverting the angle."""
     _, struct = base._flatten()
     base._unflatten((-phi,), struct)
+    
+    
+@decomposition_rule(op=qml.adjoint(qml.adjoint(qml.Hadamard(wires=0))))
+def cancel_adjoint(*params, wires, base):
+    """Decompose the adjoint of a rotation operator by inverting the angle."""
+    base.base._unflatten(*base.base._flatten())
+    
+#
+@decomposition_rule(op=qml.ctrl(qml.adjoint(qml.RX(0.2, wires=[0])), control=1))
+def flip_control_adjoint(
+    *_, wires, control_wires, control_values, work_wires, work_wire_type, base, **__
+):
+    """Decompose the control of an adjoint by applying control to the base of the adjoint
+    and taking the adjoint of the control."""
+    base_op = base.base._unflatten(*base.base._flatten())
+    qml.adjoint(
+        qml.ctrl(
+            base_op,
+            control=wires[: len(control_wires)],
+            control_values=control_values,
+            work_wires=work_wires,
+            work_wire_type=work_wire_type,
+        )
+    )
 
 
 @qml.qjit()
@@ -29,7 +53,9 @@ def circuit():
 
     qml.adjoint(qml.RY(0.432, wires=1))
 
-    adjoint_rotation(float, int)
+    adjoint_rotation(float)
+    cancel_adjoint()
+    flip_control_adjoint()
 
     return qml.expval(qml.Z(0))