Add gates.gate.CCZ

doc/source/api-reference/qibo.rst

@@ -923,6 +923,13 @@ Toffoli
     :members:
     :member-order: bysource
 
+CCZ
+"""
+
+.. autoclass:: qibo.gates.CCZ
+    :members:
+    :member-order: bysource
+
 Deutsch
 """""""
 

pyproject.toml

@@ -69,7 +69,7 @@ pylint = "^3.0.3"
 matplotlib = "^3.7.0"
 tensorflow = { version = "^2.14.1,<2.16", markers = "sys_platform == 'linux'" }
 torch = "^2.1.1"
-qibojit = { git = "https://github.com/qiboteam/qibojit.git" }
+qibojit = { git = "https://github.com/qiboteam/qibojit.git" , branch = "ccz" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 stim = "^1.12.0"
 
@@ -87,7 +87,7 @@ optional = true
 [tool.poetry.group.cuda11.dependencies]
 cupy-cuda11x = "^12.0.0"
 cuquantum-python-cu11 = "^23.3.0"
-qibojit = { git = "https://github.com/qiboteam/qibojit.git" }
+qibojit = { git = "https://github.com/qiboteam/qibojit.git" , branch = "ccz" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 
 [tool.poetry.group.cuda12]
@@ -96,7 +96,7 @@ optional = true
 [tool.poetry.group.cuda12.dependencies]
 cupy-cuda12x = "^12.0.0"
 cuquantum-python-cu12 = "^23.3.0"
-qibojit = { git = "https://github.com/qiboteam/qibojit.git" }
+qibojit = { git = "https://github.com/qiboteam/qibojit.git" , branch = "ccz" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 
 [tool.poetry.extras]

src/qibo/backends/npmatrices.py

@@ -446,6 +446,22 @@ def TOFFOLI(self):
             dtype=self.dtype,
         )
 
+    @cached_property
+    def CCZ(self):
+        return self._cast(
+            [
+                [1, 0, 0, 0, 0, 0, 0, 0],
+                [0, 1, 0, 0, 0, 0, 0, 0],
+                [0, 0, 1, 0, 0, 0, 0, 0],
+                [0, 0, 0, 1, 0, 0, 0, 0],
+                [0, 0, 0, 0, 1, 0, 0, 0],
+                [0, 0, 0, 0, 0, 1, 0, 0],
+                [0, 0, 0, 0, 0, 0, 1, 0],
+                [0, 0, 0, 0, 0, 0, 0, -1],
+            ],
+            dtype=self.dtype,
+        )
+
     def DEUTSCH(self, theta):
         sin = self.np.sin(theta) + 0j  # 0j necessary for right tensorflow dtype
         cos = self.np.cos(theta) + 0j

src/qibo/gates/gates.py

@@ -2307,6 +2307,56 @@ def congruent(self, use_toffolis: bool = True) -> List[Gate]:
         ]
 
 
+class CCZ(Gate):
+    """The controlled-CZ gate.
+
+    Corresponds to the following unitary matrix
+
+    .. math::
+        \\begin{pmatrix}
+            1 & 0 & 0 & 0 & 0 & 0 & 0 & 0 \\\\
+            0 & 1 & 0 & 0 & 0 & 0 & 0 & 0 \\\\
+            0 & 0 & 1 & 0 & 0 & 0 & 0 & 0 \\\\
+            0 & 0 & 0 & 1 & 0 & 0 & 0 & 0 \\\\
+            0 & 0 & 0 & 0 & 1 & 0 & 0 & 0 \\\\
+            0 & 0 & 0 & 0 & 0 & 1 & 0 & 0 \\\\
+            0 & 0 & 0 & 0 & 0 & 0 & 1 & 0 \\\\
+            0 & 0 & 0 & 0 & 0 & 0 & 0 & -1 \\\\
+        \\end{pmatrix}
+
+    Args:
+        q0 (int): the first control qubit id number.
+        q1 (int): the second control qubit id number.
+        q2 (int): the target qubit id number.
+    """
+
+    def __init__(self, q0, q1, q2):
+        super().__init__()
+        self.name = "ccz"
+        self.draw_label = "Z"
+        self.control_qubits = (q0, q1)
+        self.target_qubits = (q2,)
+        self.init_args = [q0, q1, q2]
+        self.unitary = True
+
+    @property
+    def qasm_label(self):
+        return "ccz"
+
+    def decompose(self) -> List[Gate]:
+        """Decomposition of :math:`\\text{CCZ}` gate.
+
+        Decompose :math:`\\text{CCZ}` gate into :class:`qibo.gates.H` in
+        the target qubit, followed by :class:`qibo.gates.TOFFOLI`, followed
+        by a :class:`qibo.gates.H` in the target qubit.
+        """
+        from qibo.transpiler.decompositions import (  # pylint: disable=C0415
+            standard_decompositions,
+        )
+
+        return standard_decompositions(self)
+
+
 class DEUTSCH(ParametrizedGate):
     """The Deutsch gate.
 

src/qibo/transpiler/decompositions.py

@@ -465,3 +465,4 @@ def _u3_to_gpi2(t, p, l):
 standard_decompositions.add(
     gates.ECR, [gates.S(0), gates.SX(1), gates.CNOT(0, 1), gates.X(0)]
 )
+standard_decompositions.add(gates.CCZ, [gates.H(2), gates.TOFFOLI(0, 1, 2), gates.H(2)])

tests/test_gates_gates.py

@@ -1206,6 +1206,39 @@ def test_toffoli(backend, applyx):
     assert gates.TOFFOLI(0, 1, 2).unitary
 
 
+def test_ccz(backend):
+    nqubits = 3
+    initial_state = random_statevector(2**nqubits, backend=backend)
+    final_state = apply_gates(
+        backend,
+        [gates.CCZ(0, 1, 2)],
+        nqubits=nqubits,
+        initial_state=initial_state,
+    )
+
+    matrix = np.array(
+        [
+            [1, 0, 0, 0, 0, 0, 0, 0],
+            [0, 1, 0, 0, 0, 0, 0, 0],
+            [0, 0, 1, 0, 0, 0, 0, 0],
+            [0, 0, 0, 1, 0, 0, 0, 0],
+            [0, 0, 0, 0, 1, 0, 0, 0],
+            [0, 0, 0, 0, 0, 1, 0, 0],
+            [0, 0, 0, 0, 0, 0, 1, 0],
+            [0, 0, 0, 0, 0, 0, 0, -1],
+        ],
+        dtype=np.complex128,
+    )
+    matrix = backend.cast(matrix, dtype=matrix.dtype)
+
+    target_state = matrix @ initial_state
+    backend.assert_allclose(final_state, target_state)
+
+    assert gates.CCZ(0, 1, 2).qasm_label == "ccz"
+    assert not gates.CCZ(0, 1, 2).clifford
+    assert gates.CCZ(0, 1, 2).unitary
+
+
 def test_deutsch(backend):
     theta = 0.1234
     nqubits = 3