Add adjoint and n_ctrls args to Operator2 primitive

.dep-versions

@@ -8,7 +8,7 @@ enzyme=v0.0.238
 
 # For a custom PL version, update the package version here and at
 # 'doc/requirements.txt'
-pennylane=0.46.0.dev38
+pennylane=0.46.0.dev46
 
 # For a custom LQ/LK version, update the package version here and at
 # 'doc/requirements.txt'

doc/releases/changelog-dev.md

@@ -10,6 +10,7 @@
   where applicable, unlocking compilation and execution for these cases.
   [(#2979)](https://github.com/PennyLaneAI/catalyst/pull/2979)
   [(#2969)](https://github.com/PennyLaneAI/catalyst/pull/2969/)
+  [(#2990)](https://github.com/PennyLaneAI/catalyst/pull/2990)
 
 * The `ResourceAnalysis` pass now reports each loop body and each subroutine as its own entry
   instead of folding their gate counts into the caller. Loops with constant bounds appear as `for_loop_<N>`

doc/requirements.txt

@@ -34,4 +34,4 @@ lxml_html_clean
 --extra-index-url https://test.pypi.org/simple/
 pennylane-lightning-kokkos==0.46.0-dev10
 pennylane-lightning==0.46.0-dev10
-pennylane==0.46.0.dev38
+pennylane==0.46.0.dev46

frontend/catalyst/from_plxpr/qfunc_interpreter.py

@@ -293,7 +293,7 @@
 
 
 @PLxPRToQuantumJaxprInterpreter.register_primitive(operator_p)
-def _handle_operator(self, *args, op_cls, hybrid_lens, hybrid_trees, **kwargs):
+def _handle_operator(self, *args, op_cls, hybrid_lens, hybrid_trees, adjoint, n_ctrls, **kwargs):
 
     if hybrid_lens or hybrid_trees or op_cls.static_argnames:
         # only support compilable_argnames for the moment
@@ -301,7 +301,7 @@
 
     wire_inputs = args[len(op_cls.dynamic_argnames) :]
     new_wires = [
         w if is_abstract_qubit(w) else qref_get_p.bind(self.init_qreg, w) for w in wire_inputs
     ]
 
     qref_operator_p.bind(
@@ -310,6 +310,8 @@
         op_cls=op_cls,
         hybrid_lens=hybrid_lens,
         hybrid_trees=hybrid_trees,
+        adjoint=adjoint,
+        n_ctrls=n_ctrls,
         **kwargs,
     )
     return []

frontend/catalyst/from_plxpr/qref_operator2_primitives.py

@@ -112,6 +112,8 @@ def _qref_operator_p_lowering(
     # will be used in future improvements
     hybrid_lens = kwargs.pop("hybrid_lens")  # pylint: disable=unused-variable
     hybrid_trees = kwargs.pop("hybrid_trees")  # pylint: disable=unused-variable
+    adjoint = kwargs.pop("adjoint")  # pylint: disable=unused-variable
+    n_ctrls = kwargs.pop("n_ctrls")  # pylint: disable=unused-variable
     wire_lens = kwargs.pop("wire_lens")
     params = args[: len(op_cls.dynamic_argnames)]
     qubits = args[len(op_cls.dynamic_argnames) :]