Add bloq counts for rotations. (#377)

quantumlib · Oct 6, 2023 · a73de04 · a73de04
1 parent 9021b71
commit a73de04
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/qualtran/bloqs/basic_gates/rotation.py b/qualtran/bloqs/basic_gates/rotation.py
@@ -14,13 +14,14 @@
 
 import abc
 from functools import cached_property
-from typing import Dict, Tuple, TYPE_CHECKING
+from typing import Dict, Optional, Set, Tuple, TYPE_CHECKING
 
 import numpy as np
 from attrs import frozen
 from cirq_ft import TComplexity
 
 from qualtran import Bloq, Signature
+from qualtran.bloqs.basic_gates.t_gate import TGate
 
 if TYPE_CHECKING:
     import cirq
@@ -46,6 +47,10 @@ def t_complexity(self):
         num_t = int(np.ceil(1.149 * np.log2(1.0 / self.eps) + 9.2))
         return TComplexity(t=num_t)
 
+    def bloq_counts(self, ssa: Optional['SympySymbolAllocator'] = None) -> Set[Tuple[int, Bloq]]:
+        num_t = int(np.ceil(1.149 * np.log2(1.0 / self.eps) + 9.2))
+        return {(num_t, TGate())}
+
 
 @frozen
 class Rz(RotationBloq):