Merge branch 'main' into vlasov

Author: charlesyuan314

dev_tools/qualtran_dev_tools/all_call_graph.py

@@ -13,6 +13,7 @@
 #  limitations under the License.
 
 """Generate the library-wide call graph from all bloq examples."""
+import logging
 import warnings
 from typing import Iterable
 
@@ -31,6 +32,8 @@
     ignore_split_join,
 )
 
+logger = logging.getLogger(__name__)
+
 
 def get_all_call_graph(bes: Iterable[BloqExample]):
     """Create a call graph that is the union of all of the bloqs in the list of bloq examples.
@@ -70,6 +73,7 @@ def keep(b: Bloq) -> bool:
     ssa = SympySymbolAllocator()
 
     for be in bes:
+        logger.info("Building call graph for: %s", be.name)
         bloq = be.make()
         _build_call_graph(
             bloq=bloq, generalizer=generalize, ssa=ssa, keep=keep, max_depth=None, g=g, depth=0

qualtran/bloqs/basic_gates/on_each.py

@@ -43,6 +43,7 @@ class OnEach(Bloq):
     Args:
         n: the number of qubits to add the bloq to.
         gate: A single qubit gate. The single qubit register must be named q.
+        target_dtype: optional dtype of the register. If not provided, default to QAny.
 
     Registers:
      - q: an n-qubit register.

qualtran/bloqs/block_encoding/lcu_block_encoding.py

@@ -115,7 +115,7 @@ def system_bitsize(self) -> int:
 
     @cached_property
     def selection_registers(self) -> Tuple[Register, ...]:
-        return self.prepare.selection_registers
+        return self.select.selection_registers
 
     @cached_property
     def junk_registers(self) -> Tuple[Register, ...]: