Fix issue where qubits are retained with draw / tracer (#1457)

* Fix issue where qubits are retained with draw / tracer --------- Signed-off-by: Alex McCaskey <amccaskey@nvidia.com>
NVIDIA · Apr 1, 2024 · f6840d0 · f6840d0
1 parent d355064
commit f6840d0
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 8 deletions.
diff --git a/python/tests/kernel/test_kernel_features.py b/python/tests/kernel/test_kernel_features.py
@@ -988,3 +988,33 @@ def kernel(argument: float):
 
     with pytest.raises(RuntimeError) as error:
         print(cudaq.draw(kernel))
+
+
+def test_draw_bug_1400():
+
+    @cudaq.kernel
+    def bell_pair():
+        q = cudaq.qvector(2)
+        h(q[0])
+        cx(q[0], q[1])
+        mz(q)
+
+
+    @cudaq.kernel
+    def kernel(angle:float):
+        q = cudaq.qubit()
+        h(q)
+        ry(angle, q)
+
+
+    print(cudaq.draw(kernel, 0.59))
+    print(cudaq.draw(kernel, 0.59))
+    circuit = cudaq.draw(bell_pair)
+    print(circuit)
+    expected_str = '''     ╭───╮     
+q0 : ┤ h ├──●──
+     ╰───╯╭─┴─╮
+q1 : ─────┤ x ├
+          ╰───╯
+'''
+    assert circuit == expected_str
diff --git a/runtime/cudaq/qis/managers/BasicExecutionManager.h b/runtime/cudaq/qis/managers/BasicExecutionManager.h
@@ -110,14 +110,6 @@ class BasicExecutionManager : public cudaq::ExecutionManager {
     if (!executionContext)
       return;
 
-    if (isInTracerMode()) {
-      for (auto &q : contextQuditIdsForDeletion)
-        returnIndex(q.id);
-
-      contextQuditIdsForDeletion.clear();
-      return;
-    }
-
     // Do any final post-processing before
     // we deallocate the qudits
     handleExecutionContextEnded();