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

emu_sv/__init__.py

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

emu_sv/dense_operator.py

@@ -138,8 +138,8 @@ class DenseOperator(Operator[complex, torch.Tensor, StateVector]):
             # and potentially user defined strings in terms of {r, g} or {0, 1}
             operators_with_tensors |= {
                 "gg": torch.tensor([[1.0, 0.0], [0.0, 0.0]], dtype=dtype),
-                "gr": torch.tensor([[0.0, 0.0], [1.0, 0.0]], dtype=dtype),
-                "rg": torch.tensor([[0.0, 1.0], [0.0, 0.0]], dtype=dtype),
+                "rg": torch.tensor([[0.0, 0.0], [1.0, 0.0]], dtype=dtype),
+                "gr": torch.tensor([[0.0, 1.0], [0.0, 0.0]], dtype=dtype),
                 "rr": torch.tensor([[0.0, 0.0], [0.0, 1.0]], dtype=dtype),
             }
         elif set(eigenstates) == {"0", "1"}:

emu_sv/state_vector.py

@@ -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

@@ -105,6 +105,11 @@ class BaseSVBackendImpl:
             raise NotImplementedError(
                 "Initial state and state preparation error can not be together."
             )
+        requested_gpu = self._config.gpu
+        if requested_gpu is None:
+            requested_gpu = True
+
+        self.resolved_gpu = requested_gpu
 
     def init_dark_qubits(self) -> None:
         if self._config.noise_model.state_prep_error > 0.0:
@@ -180,13 +185,12 @@ class SVBackendImpl(BaseSVBackendImpl):
             pulser_data: The data for the sequence to be emulated.
         """
         super().__init__(config, pulser_data)
-
         self.state: StateVector = (
-            StateVector.make(self.nqubits, gpu=self._config.gpu)
+            StateVector.make(self.nqubits, gpu=self.resolved_gpu)
             if self._config.initial_state is None
             else StateVector(
                 self._config.initial_state.vector.clone(),
-                gpu=self._config.gpu,
+                gpu=self.resolved_gpu,
             )
         )
 
@@ -222,10 +226,10 @@ class NoisySVBackendImpl(BaseSVBackendImpl):
         self.pulser_lindblads = pulser_data.lindblad_ops
 
         self.state: DensityMatrix = (
-            DensityMatrix.make(self.nqubits, gpu=self._config.gpu)
+            DensityMatrix.make(self.nqubits, gpu=self.resolved_gpu)
             if self._config.initial_state is None
             else DensityMatrix(
-                self._config.initial_state.matrix.clone(), gpu=self._config.gpu
+                self._config.initial_state.matrix.clone(), gpu=self.resolved_gpu
             )
         )
 

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,
@@ -41,8 +43,11 @@ class SVConfig(EmulationConfig):
             the size of the krylov subspace that the Lanczos algorithm maximally builds
         krylov_tolerance:
             the Lanczos algorithm uses this as the convergence tolerance
-        gpu: Use 1 gpu if True, and a GPU is available, otherwise, cpu.
-            Will cause errors if True when a gpu is not available
+        gpu: choosing the number of gpus to use during the simulation
+            - if `gpu = True`, use 1 GPU to store the state.
+            (causes errors if True when GPU not available)
+            - if `gpu = False`, use CPU to run the entire simulation.
+            - if `gpu = None` (the default value), the backend internally chooses 1 GPU.
         interaction_cutoff: Set interaction coefficients below this value to `0`.
             Potentially improves runtime and memory consumption.
         log_level: How much to log. Set to `logging.WARN` to get rid of the timestep info.
@@ -59,6 +64,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,
@@ -66,7 +73,7 @@ class SVConfig(EmulationConfig):
         dt: int = 10,
         max_krylov_dim: int = 100,
         krylov_tolerance: float = 1e-10,
-        gpu: bool = True,
+        gpu: bool | None = None,
         interaction_cutoff: float = 0.0,
         log_level: int = logging.INFO,
         log_file: pathlib.Path | None = None,

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

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

emu_sv-2.4.2.dist-info/RECORD

@@ -0,0 +1,15 @@
+emu_sv/__init__.py,sha256=HmN1DY_0fQUrgBqa3eRVIqqSg1pg11_w4FwuwepfgV4,771
+emu_sv/custom_callback_implementations.py,sha256=_7XLIDzJ-p3DVqz-Jyv0eYbl8nih2x2p-pM4cBCLumA,6367
+emu_sv/dense_operator.py,sha256=GvF0swsiFRqp83bpyaU_CXap2vm74-JLI5lHo-0Hbdk,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=-xWE30B5RI32nOG2pUR8lL3q-wufwvzxegiJexW5g4w,8952
+emu_sv/sv_config.py,sha256=xgKyEPOGzqM-jC_xJ5VYJak4CNntbp2wywbdYqCGEoM,6318
+emu_sv/time_evolution.py,sha256=Uy3qMdt3BlLB6Aq1-o5uajRTu_3fPuBCtcusHxFPPJc,13545
+emu_sv/utils.py,sha256=t0nMDVo6DF5bQW-vbsyRMCmvkyNxCU-v0Enmns9aOAU,1151
+emu_sv-2.4.2.dist-info/METADATA,sha256=RuXcAiQTi84jywPv_UkYs5OVJ0QQAGap55xNEgVSlvY,3595
+emu_sv-2.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_sv-2.4.2.dist-info/RECORD,,

emu_sv-2.4.0.dist-info/RECORD

@@ -1,15 +0,0 @@
-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=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=dkAnxEQxK-4Fn6UxhDbcb1z1EsHFUpOyVy0mjHkKFBI,9894
-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=J4yb9-Na5hAbShDySzPfbt0WXvL8JpfrTEikTK8nH9s,5943
-emu_sv/time_evolution.py,sha256=_VH4f2RF6lGKzO08WxTYJ5Aw8_pTTMRKcyMnIuxH03I,13382
-emu_sv/utils.py,sha256=t0nMDVo6DF5bQW-vbsyRMCmvkyNxCU-v0Enmns9aOAU,1151
-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,,