emu-sv 2.0.1__py3-none-any.whl → 2.0.3__py3-none-any.whl

emu_sv/__init__.py

@@ -35,4 +35,4 @@ __all__ = [
     "inner",
 ]
 
-__version__ = "2.0.1"
+__version__ = "2.0.3"

emu_sv/hamiltonian.py

@@ -50,10 +50,7 @@ class RydbergHamiltonian:
 
         self.diag: torch.Tensor = self._create_diagonal()
         self.inds = torch.tensor([1, 0], device=self.device)  # flips the state, for σˣ
-
-        self._apply_sigma_operators = self._apply_sigma_operators_real
-        if self.phis.any():
-            self._apply_sigma_operators = self._apply_sigma_operators_complex
+        self.complex = self.phis.any()
 
     def __mul__(self, vec: torch.Tensor) -> torch.Tensor:
         """
@@ -73,7 +70,10 @@ class RydbergHamiltonian:
         # (-∑ⱼΔⱼnⱼ + ∑ᵢ﹥ⱼUᵢⱼnᵢnⱼ)|ψ❭
         result = self.diag * vec
         # ∑ⱼΩⱼ/2[cos(ϕⱼ)σˣⱼ + sin(ϕⱼ)σʸⱼ]|ψ❭
-        self._apply_sigma_operators(result, vec)
+        if self.complex:
+            self._apply_sigma_operators_complex(result, vec)
+        else:
+            self._apply_sigma_operators_real(result, vec)
         return result
 
     def _apply_sigma_operators_real(

emu_sv/state_vector.py

@@ -36,6 +36,7 @@ class StateVector(State[complex, torch.Tensor]):
         vector: torch.Tensor,
         *,
         gpu: bool = True,
+        eigenstates: Sequence[Eigenstate] = ("r", "g"),
     ):
         # NOTE: this accepts also zero vectors.
 
@@ -43,6 +44,7 @@ class StateVector(State[complex, torch.Tensor]):
             len(vector)
         ).is_integer(), "The number of elements in the vector should be power of 2"
 
+        super().__init__(eigenstates=eigenstates)
         device = "cuda" if gpu and DEVICE_COUNT > 0 else "cpu"
         self.vector = vector.to(dtype=dtype, device=device)
 
@@ -68,7 +70,12 @@ class StateVector(State[complex, torch.Tensor]):
             self.vector = self.vector / norm
 
     @classmethod
-    def zero(cls, num_sites: int, gpu: bool = True) -> StateVector:
+    def zero(
+        cls,
+        num_sites: int,
+        gpu: bool = True,
+        eigenstates: Sequence[Eigenstate] = ("r", "g"),
+    ) -> StateVector:
         """
         Returns a zero uninitialized "state" vector. Warning, this has no physical meaning as-is!
 
@@ -86,7 +93,7 @@ class StateVector(State[complex, torch.Tensor]):
 
         device = "cuda" if gpu and DEVICE_COUNT > 0 else "cpu"
         vector = torch.zeros(2**num_sites, dtype=dtype, device=device)
-        return cls(vector, gpu=gpu)
+        return cls(vector, gpu=gpu, eigenstates=eigenstates)
 
     @classmethod
     def make(cls, num_sites: int, gpu: bool = True) -> StateVector:
@@ -172,7 +179,14 @@ class StateVector(State[complex, torch.Tensor]):
             The summed state
         """
         assert isinstance(other, StateVector), "`Other` state can only be a StateVector"
-        return StateVector(self.vector + other.vector, gpu=self.vector.is_cuda)
+        assert (
+            self.eigenstates == other.eigenstates
+        ), f"`Other` state has basis {other.eigenstates} != {self.eigenstates}"
+        return StateVector(
+            self.vector + other.vector,
+            gpu=self.vector.is_cuda,
+            eigenstates=self.eigenstates,
+        )
 
     def __rmul__(self, scalar: complex) -> StateVector:
         """Scalar multiplication
@@ -183,7 +197,11 @@ class StateVector(State[complex, torch.Tensor]):
         Returns:
             The scaled state
         """
-        return StateVector(scalar * self.vector, gpu=self.vector.is_cuda)
+        return StateVector(
+            scalar * self.vector,
+            gpu=self.vector.is_cuda,
+            eigenstates=self.eigenstates,
+        )
 
     def norm(self) -> torch.Tensor:
         """Returns the norm of the state
@@ -229,9 +247,6 @@ class StateVector(State[complex, torch.Tensor]):
             tensor([0.7071+0.j, 0.0000+0.j, 0.0000+0.j, 0.7071+0.j],
                    dtype=torch.complex128)
         """
-
-        # nqubits = len(next(iter(amplitudes.keys())))
-        nqubits = cls._validate_amplitudes(amplitudes=amplitudes, eigenstates=eigenstates)
         basis = set(eigenstates)
         if basis == {"r", "g"}:
             one = "r"
@@ -242,7 +257,8 @@ class StateVector(State[complex, torch.Tensor]):
         else:
             raise ValueError("Unsupported basis provided")
 
-        accum_state = StateVector.zero(num_sites=nqubits)
+        nqubits = cls._validate_amplitudes(amplitudes=amplitudes, eigenstates=eigenstates)
+        accum_state = StateVector.zero(num_sites=nqubits, eigenstates=eigenstates)
 
         for state, amplitude in amplitudes.items():
             bin_to_int = int(

emu_sv/sv_backend.py

@@ -12,7 +12,7 @@ from emu_sv.sv_config import SVConfig
 from emu_sv.time_evolution import do_time_step
 
 
-_TIME_CONVERSION_COEFF = 0.001  # Omega and delta are given in rad/ms, dt in ns
+_TIME_CONVERSION_COEFF = 0.001  # Omega and delta are given in rad/μs, dt in ns
 
 
 class SVBackend(EmulatorBackend):
@@ -90,6 +90,7 @@ class SVBackend(EmulatorBackend):
                 self.results,
             )
             self.time = time.time()
+            del H
 
         return self.results
 

{emu_sv-2.0.1.dist-info → emu_sv-2.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-sv
-Version: 2.0.1
+Version: 2.0.3
 Summary: Pasqal State Vector based pulse emulator built on PyTorch
 Project-URL: Documentation, https://pasqal-io.github.io/emulators/
 Project-URL: Repository, https://github.com/pasqal-io/emulators
@@ -25,7 +25,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.10
-Requires-Dist: emu-base==2.0.1
+Requires-Dist: emu-base==2.0.3
 Description-Content-Type: text/markdown
 
 <div align="center">

{emu_sv-2.0.1.dist-info → emu_sv-2.0.3.dist-info}/RECORD

@@ -1,13 +1,13 @@
-emu_sv/__init__.py,sha256=VYP3ZjQxPQINv-VC6URdtivwJDCAaBx1Ld5iCEdTNzw,701
+emu_sv/__init__.py,sha256=6ijDylrpDM3KZkI1hvmsh_uUAOKowLmjvxaE7wrAmOY,701
 emu_sv/custom_callback_implementations.py,sha256=zvsSiDIc56gwybKq87VFZyKsniTDye6-oFd2-R0shpg,3447
 emu_sv/dense_operator.py,sha256=NfgzVpnNitc5ZSM4RlfpAc5Ls2wFPNsTxdeFdhJSg1o,6909
 emu_sv/density_matrix_state.py,sha256=6UBLUXaJaUdzOhflrKolcnH8737JszX7sry1WmbyakI,6993
-emu_sv/hamiltonian.py,sha256=zZq-_yUVhhafHxKkTZN0hivzzWFwC9JkTEk4MV1fBag,6168
-emu_sv/state_vector.py,sha256=iX_33hGzn5ZsmG-LWE_Tjhu-Tvy4_lBOjHieT0C-VbA,9335
-emu_sv/sv_backend.py,sha256=LE8tfoo-5-o6O39ixUMZAGo-xM8m9Xwcl_HmM1pSYtw,4433
+emu_sv/hamiltonian.py,sha256=veJlJh_Q2_Fgc0IIfKPSWb6n_oem5WWGQUGDeepl924,6138
+emu_sv/state_vector.py,sha256=lqSbv4BMtDtgY0YUPuhIUNJxrlVa7vUWuN_XqwpG5sQ,9823
+emu_sv/sv_backend.py,sha256=AkEtI6-SY20D0ORro3Kv8tHDRUc8gxejSiRa6d--vBE,4452
 emu_sv/sv_config.py,sha256=QRy0VbCugmY6TQZ48nD6RxPJbpu0wzN7-E1Sud7YxLQ,5106
 emu_sv/time_evolution.py,sha256=48C0DL_SOu7Jdjk2QKBNPsevOpQlgsPYUHE7cScY-ZM,796
 emu_sv/utils.py,sha256=-axfQ2tqw0C7I9yw-28g7lytyk373DNBjDALh4kLBrM,302
-emu_sv-2.0.1.dist-info/METADATA,sha256=FSrCZuRIQ25LNbffZ2cz73nxbX-n1JbG9xgblJecYjc,3513
-emu_sv-2.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_sv-2.0.1.dist-info/RECORD,,
+emu_sv-2.0.3.dist-info/METADATA,sha256=BYetdN_TW7rMls-VeLMSObauvcKONqh8an0FZqouJfw,3513
+emu_sv-2.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_sv-2.0.3.dist-info/RECORD,,