Add support for decompositions of parameterized cirq.CCZPowGate (qu…

…antumlib#5087) * Add support for decompositions of parameterized cirq.CCZPowGate * Add test for parameterized decomposition of CCZ
rht · May 1, 2023 · d308676 · d308676
1 parent 7f53521
commit d308676
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/cirq-core/cirq/ops/three_qubit_gates.py b/cirq-core/cirq/ops/three_qubit_gates.py
@@ -95,9 +95,6 @@ def _decompose_(self, qubits):
 
         where p = T**self._exponent
         """
-        if protocols.is_parameterized(self):
-            return NotImplemented
-
         a, b, c = qubits
 
         # Hacky magic: avoid the non-adjacent edge.

diff --git a/cirq-core/cirq/ops/three_qubit_gates_test.py b/cirq-core/cirq/ops/three_qubit_gates_test.py
@@ -192,6 +192,7 @@ def test_identity_multiplication():
         (cirq.CCZ(*cirq.LineQubit.range(3)), 8),
         (cirq.CCX(*cirq.LineQubit.range(3)), 8),
         (cirq.CCZ(cirq.LineQubit(0), cirq.LineQubit(2), cirq.LineQubit(1)), 8),
+        (cirq.CCZ(cirq.LineQubit(0), cirq.LineQubit(2), cirq.LineQubit(1)) ** sympy.Symbol("s"), 8),
         (cirq.CSWAP(*cirq.LineQubit.range(3)), 9),
         (cirq.CSWAP(*reversed(cirq.LineQubit.range(3))), 9),
         (cirq.CSWAP(cirq.LineQubit(1), cirq.LineQubit(0), cirq.LineQubit(2)), 12),