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

emu_sv/__init__.py

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

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,

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-2.3.0.dist-info → emu_sv-2.4.0.dist-info}/METADATA

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

{emu_sv-2.3.0.dist-info → emu_sv-2.4.0.dist-info}/RECORD

@@ -1,15 +1,15 @@
-emu_sv/__init__.py,sha256=BWI-O-rXdn7xncmWEUdxtx7gFbWflcPD2lKcsll6t9w,771
+emu_sv/__init__.py,sha256=Oc9_UlpE7Nioggmg-pX4_8iQnh7_o9jl8n_wBTu5NJg,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/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=oZeTr44Wt1zYEmPc3yN-Zz8aKxIBTjKA0XM23T8MQkU,9871
+emu_sv/state_vector.py,sha256=dkAnxEQxK-4Fn6UxhDbcb1z1EsHFUpOyVy0mjHkKFBI,9894
 emu_sv/sv_backend.py,sha256=-soOkSEzEBK1dCKnYnbtvYjmNZtZra1_4jP3H1ROOtM,737
-emu_sv/sv_backend_impl.py,sha256=ISls8DGZNHrH3Z7kmrzk3oQU3J8lgBg2V4KG09Zt3jA,9263
+emu_sv/sv_backend_impl.py,sha256=OYsJK_DJ8dmq3QZLKiBejXrq22R5Auyye-pai07NeeY,8798
 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,,
+emu_sv-2.4.0.dist-info/METADATA,sha256=HKTJmI_Ra3dcFw2QthE2TEvEiYcvcZmrHkrFMuGY794,3595
+emu_sv-2.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_sv-2.4.0.dist-info/RECORD,,