emu-sv 2.3.0__py3-none-any.whl → 2.4.1__py3-none-any.whl

emu_sv/__init__.py

@@ -38,4 +38,4 @@ __all__ = [
     "DensityMatrix",
 ]
 
-__version__ = "2.3.0"
+__version__ = "2.4.1"

emu_sv/density_matrix_state.py

@@ -80,7 +80,9 @@ class DensityMatrix(State[complex, torch.Tensor]):
             self.matrix.shape == other.matrix.shape
         ), "States do not have the same number of sites"
 
-        return torch.vdot(self.matrix.flatten(), other.matrix.flatten())
+        return torch.vdot(
+            self.matrix.flatten(), other.matrix.to(self.matrix.device).flatten()
+        )
 
     @classmethod
     def from_state_vector(cls, state: StateVector) -> DensityMatrix:

emu_sv/state_vector.py

@@ -134,7 +134,7 @@ class StateVector(State[complex, torch.Tensor]):
         ), "States do not have the same shape"
 
         # by our internal convention inner and norm return to cpu
-        return torch.vdot(self.vector, other.vector).cpu()
+        return torch.vdot(self.vector, other.vector.to(self.vector.device)).cpu()
 
     def sample(
         self,
@@ -275,7 +275,7 @@ class StateVector(State[complex, torch.Tensor]):
         return accum_state, amplitudes
 
     def overlap(self, other: StateVector, /) -> torch.Tensor:
-        return self.inner(other)
+        return torch.abs(self.inner(other)) ** 2
 
 
 def inner(left: StateVector, right: StateVector) -> torch.Tensor:

emu_sv/sv_backend_impl.py

@@ -1,16 +1,14 @@
 from abc import abstractmethod
 import time
 import typing
+import torch
 
 from emu_sv.hamiltonian import RydbergHamiltonian
 from emu_sv.lindblad_operator import RydbergLindbladian
 from pulser import Sequence
-import torch
-from resource import RUSAGE_SELF, getrusage
 
 from pulser.backend import Results, Observable, State, EmulationConfig
-from emu_base import PulserData
-from emu_base.noise import pick_dark_qubits
+from emu_base import PulserData, get_max_rss
 
 from emu_sv.state_vector import StateVector
 from emu_sv.density_matrix_state import DensityMatrix
@@ -46,20 +44,12 @@ class Statistics(Observable):
         assert isinstance(state, StateVector | DensityMatrix)
         assert isinstance(config, SVConfig)
         duration = self.data[-1]
-        if isinstance(state, StateVector) and state.vector.is_cuda:
-            max_mem_per_device = (
-                torch.cuda.max_memory_allocated(device) * 1e-6
-                for device in range(torch.cuda.device_count())
-            )
-            max_mem = max(max_mem_per_device)
-        elif isinstance(state, DensityMatrix) and state.matrix.is_cuda:
-            max_mem_per_device = (
-                torch.cuda.max_memory_allocated(device) * 1e-6
-                for device in range(torch.cuda.device_count())
-            )
-            max_mem = max(max_mem_per_device)
-        else:
-            max_mem = getrusage(RUSAGE_SELF).ru_maxrss * 1e-3
+        max_mem = get_max_rss(
+            isinstance(state, StateVector)
+            and state.vector.is_cuda
+            or isinstance(state, DensityMatrix)
+            and state.matrix.is_cuda
+        )
 
         config.logger.info(
             f"step = {len(self.data)}/{self.timestep_count}, "
@@ -118,8 +108,9 @@ class BaseSVBackendImpl:
 
     def init_dark_qubits(self) -> None:
         if self._config.noise_model.state_prep_error > 0.0:
-            self.well_prepared_qubits_filter = pick_dark_qubits(
-                self._config.noise_model.state_prep_error, self.nqubits
+            bad_atoms = self._pulser_data.hamiltonian.bad_atoms
+            self.well_prepared_qubits_filter = torch.tensor(
+                [bool(bad_atoms[x]) for x in self._pulser_data.qubit_ids]
             )
         else:
             self.well_prepared_qubits_filter = None

emu_sv/sv_config.py

@@ -6,6 +6,8 @@ from types import MethodType
 from typing import Any, ClassVar
 
 from emu_sv.utils import choose
+from emu_sv.state_vector import StateVector
+from emu_sv.dense_operator import DenseOperator
 
 from emu_sv.custom_callback_implementations import (
     qubit_occupation_sv_impl,
@@ -59,6 +61,8 @@ class SVConfig(EmulationConfig):
 
     # Whether to warn if unexpected kwargs are received
     _enforce_expected_kwargs: ClassVar[bool] = True
+    _state_type = StateVector
+    _operator_type = DenseOperator
 
     def __init__(
         self,

emu_sv/time_evolution.py

@@ -140,7 +140,10 @@ class EvolveStateVector(torch.autograd.Function):
             interaction_matrix=interaction_matrix,
             device=state.device,
         )
-        op = lambda x: -1j * dt * (ham * x)
+
+        def op(x: torch.Tensor) -> torch.Tensor:
+            return -1j * dt * (ham * x)
+
         res = krylov_exp(
             op,
             state,
@@ -273,7 +276,10 @@ class EvolveStateVector(torch.autograd.Function):
         )
 
         if any(ctx.needs_input_grad[1:5]):
-            op = lambda x: -1j * dt * (ham * x)
+
+            def op(x: torch.Tensor):
+                return -1j * dt * (ham * x)
+
             lanczos_vectors_state, dS, lanczos_vectors_grad = double_krylov(
                 op, state, grad_state_out, tolerance
             )
@@ -316,7 +322,10 @@ class EvolveStateVector(torch.autograd.Function):
                     grad_int_mat[i, j] = (-1j * dt * torch.tensordot(Vg.conj(), v)).real
 
         if ctx.needs_input_grad[5]:
-            op = lambda x: (1j * dt) * (ham * x)
+
+            def op(x: torch.Tensor):
+                return (1j * dt) * (ham * x)
+
             grad_state_in = krylov_exp(op, grad_state_out.detach(), tolerance, tolerance)
 
         return (
@@ -353,7 +362,9 @@ class EvolveDensityMatrix:
             device=density_matrix.device,
         )
 
-        op = lambda x: -1j * dt * (ham @ x)
+        def op(x: torch.Tensor) -> torch.Tensor:
+            return -1j * dt * (ham @ x)
+
         return (
             krylov_exp(
                 op,

{emu_sv-2.3.0.dist-info → emu_sv-2.4.1.dist-info}/METADATA

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

emu_sv-2.4.1.dist-info/RECORD

@@ -0,0 +1,15 @@
+emu_sv/__init__.py,sha256=Sv7BbiGYuZzUh30AqMZpyA9m2zM8z-UKh_D2C6LACNs,771
+emu_sv/custom_callback_implementations.py,sha256=_7XLIDzJ-p3DVqz-Jyv0eYbl8nih2x2p-pM4cBCLumA,6367
+emu_sv/dense_operator.py,sha256=AvgntJNwwtf3Wl66CIWSwUezVYV3vignCL24SQQQwQg,5901
+emu_sv/density_matrix_state.py,sha256=6QmLZvqEHLR64r0nD7D2jZIiAYOgciNVCjh3ywfvIs0,7243
+emu_sv/hamiltonian.py,sha256=CqNGuWJlO2ZljK47wt130s-5uKiOldQUsC3tjwk1mKA,6106
+emu_sv/lindblad_operator.py,sha256=pgjRNLBcvEM2-qxM8uy9wL74OtrD4A8trQeERi_AXH8,8892
+emu_sv/state_vector.py,sha256=Ookw_BDqzs9OFppQd7kaMxDSkaj5F_clhOCgo-LSaMg,9910
+emu_sv/sv_backend.py,sha256=-soOkSEzEBK1dCKnYnbtvYjmNZtZra1_4jP3H1ROOtM,737
+emu_sv/sv_backend_impl.py,sha256=OYsJK_DJ8dmq3QZLKiBejXrq22R5Auyye-pai07NeeY,8798
+emu_sv/sv_config.py,sha256=6fcuC9RKc7Pp55Yb-lWCGtZ7UJP53oLyZsy1f9wBtRg,6100
+emu_sv/time_evolution.py,sha256=Uy3qMdt3BlLB6Aq1-o5uajRTu_3fPuBCtcusHxFPPJc,13545
+emu_sv/utils.py,sha256=t0nMDVo6DF5bQW-vbsyRMCmvkyNxCU-v0Enmns9aOAU,1151
+emu_sv-2.4.1.dist-info/METADATA,sha256=5nUzMJNramhABPKHG1D7h78W6L2MeSJpkZg1wAmDg2c,3595
+emu_sv-2.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_sv-2.4.1.dist-info/RECORD,,

emu_sv-2.3.0.dist-info/RECORD

@@ -1,15 +0,0 @@
-emu_sv/__init__.py,sha256=BWI-O-rXdn7xncmWEUdxtx7gFbWflcPD2lKcsll6t9w,771
-emu_sv/custom_callback_implementations.py,sha256=_7XLIDzJ-p3DVqz-Jyv0eYbl8nih2x2p-pM4cBCLumA,6367
-emu_sv/dense_operator.py,sha256=AvgntJNwwtf3Wl66CIWSwUezVYV3vignCL24SQQQwQg,5901
-emu_sv/density_matrix_state.py,sha256=BJpFzxjgWQOb_D6dehdA6dRUUQKP1iVccTDrZKicwRE,7198
-emu_sv/hamiltonian.py,sha256=CqNGuWJlO2ZljK47wt130s-5uKiOldQUsC3tjwk1mKA,6106
-emu_sv/lindblad_operator.py,sha256=pgjRNLBcvEM2-qxM8uy9wL74OtrD4A8trQeERi_AXH8,8892
-emu_sv/state_vector.py,sha256=oZeTr44Wt1zYEmPc3yN-Zz8aKxIBTjKA0XM23T8MQkU,9871
-emu_sv/sv_backend.py,sha256=-soOkSEzEBK1dCKnYnbtvYjmNZtZra1_4jP3H1ROOtM,737
-emu_sv/sv_backend_impl.py,sha256=ISls8DGZNHrH3Z7kmrzk3oQU3J8lgBg2V4KG09Zt3jA,9263
-emu_sv/sv_config.py,sha256=J4yb9-Na5hAbShDySzPfbt0WXvL8JpfrTEikTK8nH9s,5943
-emu_sv/time_evolution.py,sha256=_VH4f2RF6lGKzO08WxTYJ5Aw8_pTTMRKcyMnIuxH03I,13382
-emu_sv/utils.py,sha256=t0nMDVo6DF5bQW-vbsyRMCmvkyNxCU-v0Enmns9aOAU,1151
-emu_sv-2.3.0.dist-info/METADATA,sha256=yc7Z-DlF125zrnUeErZ24WrRnFKjbyPxeNAfz6RwjDI,3595
-emu_sv-2.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_sv-2.3.0.dist-info/RECORD,,