Decompose Type Error

Author: mpharrigan

qualtran/bloqs/arithmetic/comparison.py

@@ -1087,6 +1087,8 @@ def build_composite_bloq(
             a = bb.add(SignExtend(self.dtype, QInt(self.dtype.bitsize + 1)), x=a)
             b = bb.add(SignExtend(self.dtype, QInt(self.dtype.bitsize + 1)), x=b)
         else:
+            if self.dtype.is_symbolic():
+                raise DecomposeTypeError(f"Cannot decompose symbolic {self}")
             a = bb.join(np.concatenate([[bb.allocate(1)], bb.split(a)]))
             b = bb.join(np.concatenate([[bb.allocate(1)], bb.split(b)]))
 

qualtran/bloqs/arithmetic/controlled_addition.py

@@ -23,6 +23,7 @@
     bloq_example,
     BloqBuilder,
     BloqDocSpec,
+    DecomposeTypeError,
     QBit,
     QInt,
     QUInt,
@@ -135,6 +136,9 @@ def wire_symbol(self, soq: 'Soquet') -> 'WireSymbol':
     def build_composite_bloq(
         self, bb: 'BloqBuilder', ctrl: 'Soquet', a: 'Soquet', b: 'Soquet'
     ) -> Dict[str, 'SoquetT']:
+        if self.a_dtype.is_symbolic() or self.b_dtype.is_symbolic():
+            raise DecomposeTypeError(f"Cannot support symbolic {self}")
+
         a_arr = bb.split(a)
         ctrl_q = bb.split(ctrl)[0]
         ancilla_arr = []

qualtran/bloqs/data_loading/select_swap_qrom.py

@@ -22,7 +22,15 @@
 import sympy
 from numpy.typing import ArrayLike
 
-from qualtran import bloq_example, BloqDocSpec, BoundedQUInt, GateWithRegisters, Register, Signature
+from qualtran import (
+    bloq_example,
+    BloqDocSpec,
+    BoundedQUInt,
+    DecomposeTypeError,
+    GateWithRegisters,
+    Register,
+    Signature,
+)
 from qualtran.bloqs.arithmetic.bitwise import Xor
 from qualtran.bloqs.bookkeeping import Partition
 from qualtran.bloqs.data_loading.qrom import QROM
@@ -394,7 +402,7 @@ def build_composite_bloq(self, bb: 'BloqBuilder', **soqs: 'SoquetT') -> Dict[str
         target = [soqs.pop(reg.name) for reg in self.target_registers]
         # Allocate intermediate clean/dirty ancilla for the underlying QROM call.
         if is_symbolic(*self.block_sizes):
-            raise ValueError(
+            raise DecomposeTypeError(
                 f"Cannot decompose SelectSwapQROM bloq with symbolic block sizes. Found {self.block_sizes=}"
             )
         block_sizes = cast(Tuple[int, ...], self.block_sizes)

qualtran/bloqs/mcmt/and_bloq.py

@@ -385,6 +385,8 @@ def decompose_from_registers(
         yield self._decompose_via_tree(control, self.concrete_cvs, ancilla, *target)
 
     def decompose_bloq(self) -> 'CompositeBloq':
+        if is_symbolic(self.cvs):
+            raise DecomposeTypeError(f"Cannot decompose symbolic {self}.")
         return decompose_from_cirq_style_method(self)
 
     def wire_symbol(self, reg: Optional[Register], idx: Tuple[int, ...] = tuple()) -> 'WireSymbol':

qualtran/bloqs/mod_arithmetic/mod_addition.py

@@ -23,6 +23,7 @@
     Bloq,
     bloq_example,
     BloqDocSpec,
+    DecomposeTypeError,
     GateWithRegisters,
     QBit,
     QMontgomeryUInt,
@@ -89,7 +90,7 @@ def on_classical_vals(
 
     def build_composite_bloq(self, bb: 'BloqBuilder', x: Soquet, y: Soquet) -> Dict[str, 'SoquetT']:
         if is_symbolic(self.bitsize):
-            raise NotImplementedError(f'symbolic decomposition is not supported for {self}')
+            raise DecomposeTypeError(f'Symbolic decomposition is not supported for {self}')
         # Allocate ancilla bits for use in addition.
         junk_bit = bb.allocate(n=1)
         sign = bb.allocate(n=1)
@@ -390,6 +391,9 @@ def on_classical_vals(
     def build_composite_bloq(
         self, bb: 'BloqBuilder', ctrl, x: Soquet, y: Soquet
     ) -> Dict[str, 'SoquetT']:
+        if self.dtype.is_symbolic():
+            raise DecomposeTypeError(f"Cannot decompose symbolic {self}")
+
         y_arr = bb.split(y)
         ancilla = bb.allocate(1)
         x = bb.add(Cast(self.dtype, QUInt(self.dtype.bitsize)), reg=x)