Don't use black box QROAM in THC call graph. (#614)

Don't use black box in THC call graph.
quantumlib · Feb 2, 2024 · 2c7bbed · 2c7bbed
1 parent 79c324d
commit 2c7bbed
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/qualtran/bloqs/chemistry/thc/prepare.py b/qualtran/bloqs/chemistry/thc/prepare.py
@@ -39,7 +39,6 @@
 )
 from qualtran.bloqs.basic_gates import Hadamard, Ry, Toffoli, XGate
 from qualtran.bloqs.basic_gates.swap import CSwap
-from qualtran.bloqs.chemistry.black_boxes import QROAM
 from qualtran.bloqs.multi_control_multi_target_pauli import MultiControlPauli
 from qualtran.bloqs.on_each import OnEach
 from qualtran.bloqs.reflection import Reflection
@@ -435,8 +434,11 @@ def build_call_graph(self, ssa: 'SympySymbolAllocator') -> Set['BloqCountT']:
         data_size = self.num_spin_orb // 2 + self.num_mu * (self.num_mu + 1) // 2
         nd = (data_size - 1).bit_length()
         cost_2 = (ToContiguousIndex(nmu, nd), 1)
-        m = 2 * nmu + 2 + self.keep_bitsize
-        cost_3 = (QROAM(data_size, m), 1)
+        qroam = SelectSwapQROM(
+            *(self.theta, self.alt_theta, self.alt_mu, self.alt_nu, self.keep),
+            target_bitsizes=(1, 1, nmu, nmu, self.keep_bitsize),
+        )
+        cost_3 = (qroam, 1)
         cost_4 = (OnEach(self.keep_bitsize, Hadamard()), 1)
         cost_5 = (LessThanEqual(self.keep_bitsize, self.keep_bitsize), 2)
         cost_6 = (CSwap(nmu), 3)