Adapt Bernstein Vazirani notebook to uncomputation

Author: ronhass-classiq

algorithms/bernstein_vazirani/bernstein_vazirani.ipynb

@@ -141,12 +141,12 @@
     "@qfunc\n",
     "def bv_function(a: CInt, x: QArray):\n",
     "    aux = QBit()\n",
+    "    allocate(aux)\n",
     "    within_apply(\n",
     "        lambda: hadamard_transform(x),\n",
-    "        lambda: within_apply(\n",
-    "            lambda: (allocate(aux), X(aux), H(aux)), lambda: bv_predicate(a, x, aux)\n",
-    "        ),\n",
-    "    )"
+    "        lambda: within_apply(lambda: (X(aux), H(aux)), lambda: bv_predicate(a, x, aux)),\n",
+    "    )\n",
+    "    free(aux)"
    ]
   },
   {

algorithms/bernstein_vazirani/bernstein_vazirani.qmod

@@ -8,17 +8,18 @@ qperm bv_predicate(a: int, const x: qbit[], res: qbit) {
 
 qfunc bv_function(a: int, x: qbit[]) {
   aux: qbit;
+  allocate(aux);
   within {
     hadamard_transform(x);
   } apply {
     within {
-      allocate(aux);
       X(aux);
       H(aux);
     } apply {
       bv_predicate(a, x, aux);
     }
   }
+  free(aux);
 }
 
 qfunc main(output x: qnum<5>) {