implemented dynamic mcp

Author: positr0nium

src/qrisp/alg_primitives/mcx_algs/balauca.py

@@ -488,12 +488,12 @@ def jasp_balauca_mcx(ctrls, target, ctrl_state):
 def jasp_balauca_mcp(phi, ctrls, ctrl_state):
 
     from qrisp import mcx, QuantumBool, cp, p
+    N = jlen(ctrls)
+    
     ctrl_state = jnp.int64(ctrl_state)
-    ctrl_state = cond(ctrl_state == -1, lambda x : x + 2**ctrls.size, lambda x : x, ctrl_state)
+    ctrl_state = cond(ctrl_state == -1, lambda x : x + 2**N, lambda x : x, ctrl_state)
     target = QuantumBool()
 
-    N = ctrls.size
-    
     with conjugate(ctrl_state_conjugator)(ctrls, ctrl_state):
 
         with control(N == 1):

src/qrisp/core/gate_application_functions.py

@@ -707,7 +707,7 @@ def mcp(phi, qubits, method="auto", ctrl_state=-1):
 
     """
 
-    from qrisp.alg_primitives.mcx_algs import hybrid_mcx
+    from qrisp.alg_primitives.mcx_algs import hybrid_mcx, jasp_balauca_mcp
     from qrisp import QuantumBool
     from qrisp.misc import bin_rep, gate_wrap
     import numpy as np
@@ -726,6 +726,10 @@ def balauca_mcp(phi, qubits, ctrl_state):
 
         temp.delete()
 
+    if check_for_tracing_mode():
+        jasp_balauca_mcp(phi, qubits, ctrl_state)
+        return
+
     n = len(qubits)
 
     if not isinstance(ctrl_state, str):

tests/jax_tests/test_dynamic_mcx.py

@@ -57,5 +57,43 @@ def main(j):
                 assert k[1]
             else:
                 assert not k[1]
+                
+    # Test dynamic mcp
+        
+    @jaspify
+    def main(phi, i):
+        
+        qv = QuantumFloat(i)
+    
+        x(qv[:qv.size-1])
+        
+        with conjugate(h)(qv[qv.size-1]):
+            mcp(phi, qv)
+        
+        return measure(qv)
+            
+    assert main(np.pi, 5) == 31
 
-        
+    @jaspify
+    def main(phi, i, j):
+        
+        qv = QuantumFloat(i)
+    
+        with conjugate(h)(qv[qv.size-1]):
+            mcp(phi, qv, ctrl_state = j)
+        
+        return measure(qv)
+            
+    assert main(np.pi, 5, 0) == 16        
+    
+    @jaspify
+    def main(phi, i, j):
+        
+        qv = QuantumFloat(i)
+    
+        with conjugate(h)(qv[qv.size-1]):
+            mcp(phi, [qv[i] for i in range(5)], ctrl_state = j)
+        
+        return measure(qv)
+            
+    assert main(np.pi, 5, 0) == 16