emu-sv 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

emu_sv/__init__.py

@@ -39,4 +39,4 @@ __all__ = [
 ]
 
 
-__version__ = "1.0.0"
+__version__ = "1.0.1"

emu_sv/hamiltonian.py

@@ -71,15 +71,14 @@ class RydbergHamiltonian:
             the resulting state vector.
         """
         # (-∑ⱼΔⱼnⱼ + ∑ᵢ﹥ⱼUᵢⱼnᵢnⱼ)|ψ❭
-        diag_result = self.diag * vec
+        result = self.diag * vec
         # ∑ⱼΩⱼ/2[cos(ϕⱼ)σˣⱼ + sin(ϕⱼ)σʸⱼ]|ψ❭
-        sigma_result = self._apply_sigma_operators(vec)
-        result: torch.Tensor
-        result = diag_result + sigma_result
-
+        self._apply_sigma_operators(result, vec)
         return result
 
-    def _apply_sigma_operators_real(self, vec: torch.Tensor) -> torch.Tensor:
+    def _apply_sigma_operators_real(
+        self, result: torch.Tensor, vec: torch.Tensor
+    ) -> None:
         """
         Apply the ∑ⱼ(Ωⱼ/2)σˣⱼ operator to the input vector |ψ❭.
 
@@ -89,8 +88,6 @@ class RydbergHamiltonian:
         Returns:
             the resulting state vector.
         """
-        result = torch.zeros_like(vec)
-
         dim_to_act = 1
         for n, omega_n in enumerate(self.omegas):
             shape_n = (2**n, 2, 2 ** (self.nqubits - n - 1))
@@ -98,9 +95,9 @@ class RydbergHamiltonian:
             result = result.reshape(shape_n)
             result.index_add_(dim_to_act, self.inds, vec, alpha=omega_n)
 
-        return result.reshape(-1)
-
-    def _apply_sigma_operators_complex(self, vec: torch.Tensor) -> torch.Tensor:
+    def _apply_sigma_operators_complex(
+        self, result: torch.Tensor, vec: torch.Tensor
+    ) -> None:
         """
         Apply the ∑ⱼΩⱼ/2[cos(ϕⱼ)σˣⱼ + sin(ϕⱼ)σʸⱼ] operator to the input vector |ψ❭.
 
@@ -111,7 +108,6 @@ class RydbergHamiltonian:
             the resulting state vector.
         """
         c_omegas = self.omegas * torch.exp(1j * self.phis)
-        result = torch.zeros_like(vec)
 
         dim_to_act = 1
         for n, c_omega_n in enumerate(c_omegas):
@@ -128,8 +124,6 @@ class RydbergHamiltonian:
                 alpha=c_omega_n.conj(),
             )
 
-        return result.reshape(-1)
-
     def _create_diagonal(self) -> torch.Tensor:
         """
         Return the diagonal elements of the Rydberg Hamiltonian matrix

emu_sv/state_vector.py

@@ -127,7 +127,6 @@ class StateVector(State):
         """
 
         probabilities = torch.abs(self.vector) ** 2
-        probabilities /= probabilities.sum()  # multinomial does not normalize the input
 
         outcomes = torch.multinomial(probabilities, num_shots, replacement=True)
 

emu_sv/sv_backend.py

@@ -9,6 +9,9 @@ import torch
 from time import time
 from resource import RUSAGE_SELF, getrusage
 from emu_base import DEVICE_COUNT
+from copy import deepcopy
+
+_TIME_CONVERSION_COEFF = 0.001  # Omega and delta are given in rad/ms, dt in ns
 
 
 class SVBackend(Backend):
@@ -36,24 +39,24 @@ class SVBackend(Backend):
         data = PulserData(sequence=sequence, config=sv_config, dt=sv_config.dt)
         omega, delta, phi = data.omega, data.delta, data.phi
 
+        target_times = data.target_times
+
         nsteps = omega.shape[0]
         nqubits = omega.shape[1]
         device = "cuda" if sv_config.gpu and DEVICE_COUNT > 0 else "cpu"
 
         if sv_config.initial_state is not None:
-            state = sv_config.initial_state
+            state = deepcopy(sv_config.initial_state)
             state.vector = state.vector.to(device)
         else:
             state = StateVector.make(nqubits, gpu=sv_config.gpu)
 
-        dt = sv_config.dt * 1e-3  # ns to µs
-
         for step in range(nsteps):
-
             start = time()
+            dt = target_times[step + 1] - target_times[step]
 
             state.vector, H = do_time_step(
-                dt,
+                dt * _TIME_CONVERSION_COEFF,
                 omega[step],
                 delta[step],
                 phi[step],
@@ -65,7 +68,7 @@ class SVBackend(Backend):
             for callback in sv_config.callbacks:
                 callback(
                     sv_config,
-                    (step + 1) * sv_config.dt,
+                    target_times[step + 1],
                     state,
                     H,  # type: ignore[arg-type]
                     results,

{emu_sv-1.0.0.dist-info → emu_sv-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-sv
-Version: 1.0.0
+Version: 1.0.1
 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==1.2.6
+Requires-Dist: emu-base==1.2.7
 Description-Content-Type: text/markdown
 
 <div align="center">

emu_sv-1.0.1.dist-info/RECORD

@@ -0,0 +1,11 @@
+emu_sv/__init__.py,sha256=rcIsdgWgk5n75z0fSQtQsVgrFs_5jtA52WwGsTxDgAs,845
+emu_sv/custom_callback_implementations.py,sha256=H7fy0trUtY00wid6VeRJ2inUazFbkMlgavTfm8QtGq8,2754
+emu_sv/dense_operator.py,sha256=69rv1J5jsHSRoPsgZqKJZnttCgMLIk4tDCBsOaOBVR8,7034
+emu_sv/hamiltonian.py,sha256=Rg_tLGTgaL4Y8xWtO1P2axRQNrdKfHRnHwnv_BfZTEI,6091
+emu_sv/state_vector.py,sha256=s26ktVkuBID9IU6ySMup8ci7M5Tn6HHCT4jcyZGxV_w,8036
+emu_sv/sv_backend.py,sha256=lIi_aTyfW-pzrfxwJHSoF5klwJnWtLv3awwU4DSL4m4,3770
+emu_sv/sv_config.py,sha256=NUAxqYG0NTWbarVreFr1Tb-8FzB0LVQ0fYnb2QSCzxo,3161
+emu_sv/time_evolution.py,sha256=48C0DL_SOu7Jdjk2QKBNPsevOpQlgsPYUHE7cScY-ZM,796
+emu_sv-1.0.1.dist-info/METADATA,sha256=hNCpmCxzB_i_ciIRzqebQ9seyPhp8F0DiuceBYG4T7k,3513
+emu_sv-1.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_sv-1.0.1.dist-info/RECORD,,

emu_sv-1.0.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-emu_sv/__init__.py,sha256=Ji7PePRQka8XrVMw3N3i2k_sBwMQIj6LBOZFGZ0HNvs,845
-emu_sv/custom_callback_implementations.py,sha256=H7fy0trUtY00wid6VeRJ2inUazFbkMlgavTfm8QtGq8,2754
-emu_sv/dense_operator.py,sha256=69rv1J5jsHSRoPsgZqKJZnttCgMLIk4tDCBsOaOBVR8,7034
-emu_sv/hamiltonian.py,sha256=uzR7XHsv0QXBbcNuk93-phrO-On-WfJCINWz7Zofc90,6270
-emu_sv/state_vector.py,sha256=XsbHDNu6LDBop_qVVUoaw8blVPOystP8MOQ9demME6g,8125
-emu_sv/sv_backend.py,sha256=cghNpJq8ALdcoHCnx0xHcZBM9x8URLRFo0_sZ5CXXvY,3576
-emu_sv/sv_config.py,sha256=NUAxqYG0NTWbarVreFr1Tb-8FzB0LVQ0fYnb2QSCzxo,3161
-emu_sv/time_evolution.py,sha256=48C0DL_SOu7Jdjk2QKBNPsevOpQlgsPYUHE7cScY-ZM,796
-emu_sv-1.0.0.dist-info/METADATA,sha256=yqdJOFSlS5RqKiBC-Tx_LSL2n3hFnkt1unkv3UMNJG8,3513
-emu_sv-1.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_sv-1.0.0.dist-info/RECORD,,