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

emu_sv/__init__.py

@@ -38,4 +38,4 @@ __all__ = [
     "DensityMatrix",
 ]
 
-__version__ = "2.4.1"
+__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/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

@@ -43,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.
@@ -70,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-2.4.1.dist-info → emu_sv-2.4.2.dist-info}/METADATA

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

{emu_sv-2.4.1.dist-info → emu_sv-2.4.2.dist-info}/RECORD

@@ -1,15 +1,15 @@
-emu_sv/__init__.py,sha256=Sv7BbiGYuZzUh30AqMZpyA9m2zM8z-UKh_D2C6LACNs,771
+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=AvgntJNwwtf3Wl66CIWSwUezVYV3vignCL24SQQQwQg,5901
+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=OYsJK_DJ8dmq3QZLKiBejXrq22R5Auyye-pai07NeeY,8798
-emu_sv/sv_config.py,sha256=6fcuC9RKc7Pp55Yb-lWCGtZ7UJP53oLyZsy1f9wBtRg,6100
+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.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.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,,