emu-base 1.2.5__py3-none-any.whl → 1.2.6__py3-none-any.whl

emu_base/__init__.py

@@ -15,6 +15,7 @@ from .base_classes.default_callbacks import (
     StateResult,
     SecondMomentOfEnergy,
 )
+from .constants import DEVICE_COUNT
 from .pulser_adapter import PulserData, HamiltonianType
 from .math.brents_root_finding import find_root_brents
 from .math.krylov_exp import krylov_exp, DEFAULT_MAX_KRYLOV_DIM
@@ -42,6 +43,7 @@ __all__ = [
     "krylov_exp",
     "HamiltonianType",
     "DEFAULT_MAX_KRYLOV_DIM",
+    "DEVICE_COUNT",
 ]
 
-__version__ = "1.2.5"
+__version__ = "1.2.6"

emu_base/constants.py

@@ -0,0 +1,4 @@
+import torch
+
+
+DEVICE_COUNT = torch.cuda.device_count()

emu_base/pulser_adapter.py

@@ -159,23 +159,44 @@ def _extract_omega_delta_phi(
 
     step = 0
     t = (step + 1 / 2) * dt
+    omega_1 = torch.zeros_like(omega[0])
+    omega_2 = torch.zeros_like(omega[0])
 
     while t < max_duration:
-        for q_pos, q_id in enumerate(sequence.register.qubit_ids):
-            omega[step, q_pos] = (
-                locals_a_d_p[q_id]["amp"][math.floor(t)]
-                + locals_a_d_p[q_id]["amp"][math.ceil(t)]
-            ) / 2.0
-            delta[step, q_pos] = (
-                locals_a_d_p[q_id]["det"][math.floor(t)]
-                + locals_a_d_p[q_id]["det"][math.ceil(t)]
-            ) / 2.0
-            phi[step, q_pos] = (
-                locals_a_d_p[q_id]["phase"][math.floor(t)]
-                + locals_a_d_p[q_id]["phase"][math.ceil(t)]
-            ) / 2.0
-        if t in global_times:
-            omega[step] *= waist_factors
+        # The sampled values correspond to the start of each interval
+        # To maximize the order of the solver, we need the values in the middle
+        if math.ceil(t) < max_duration:
+            # If we're not the final step, approximate this using linear interpolation
+            # Note that for dt even, t1=t2
+            for q_pos, q_id in enumerate(sequence.register.qubit_ids):
+                t1 = math.floor(t)
+                t2 = math.ceil(t)
+                omega_1[q_pos] = locals_a_d_p[q_id]["amp"][t1]
+                omega_2[q_pos] = locals_a_d_p[q_id]["amp"][t2]
+                delta[step, q_pos] = (
+                    locals_a_d_p[q_id]["det"][t1] + locals_a_d_p[q_id]["det"][t2]
+                ) / 2.0
+                phi[step, q_pos] = (
+                    locals_a_d_p[q_id]["phase"][t1] + locals_a_d_p[q_id]["phase"][t2]
+                ) / 2.0
+            # omegas at different times need to have the laser waist applied independently
+            if t1 in global_times:
+                omega_1 *= waist_factors
+            if t2 in global_times:
+                omega_2 *= waist_factors
+            omega[step] = 0.5 * (omega_1 + omega_2)
+        else:
+            # We're in the final step and dt=1, approximate this using linear extrapolation
+            # we can reuse omega_1 and omega_2 from before
+            for q_pos, q_id in enumerate(sequence.register.qubit_ids):
+                delta[step, q_pos] = (
+                    3.0 * locals_a_d_p[q_id]["det"][t2] - locals_a_d_p[q_id]["det"][t1]
+                ) / 2.0
+                phi[step, q_pos] = (
+                    3.0 * locals_a_d_p[q_id]["phase"][t2]
+                    - locals_a_d_p[q_id]["phase"][t1]
+                ) / 2.0
+            omega[step] = torch.clamp(0.5 * (3 * omega_2 - omega_1).real, min=0.0)
         step += 1
         t = (step + 1 / 2) * dt
 
@@ -250,11 +271,12 @@ class PulserData:
         ] = 0.0
         self.masked_interaction_matrix = self.full_interaction_matrix.clone()
 
-        slm_targets = sequence._slm_mask_targets
         self.slm_end_time = (
             sequence._slm_mask_time[1] if len(sequence._slm_mask_time) > 1 else 0.0
         )
 
-        for target in slm_targets:
+        # disable interaction for SLM masked qubits
+        slm_targets = list(sequence._slm_mask_targets)
+        for target in sequence.register.find_indices(slm_targets):
             self.masked_interaction_matrix[target] = 0.0
             self.masked_interaction_matrix[:, target] = 0.0

{emu_base-1.2.5.dist-info → emu_base-1.2.6.dist-info}/METADATA

@@ -1,7 +1,10 @@
 Metadata-Version: 2.4
 Name: emu-base
-Version: 1.2.5
+Version: 1.2.6
 Summary: Pasqal base classes for emulators
+Project-URL: Documentation, https://pasqal-io.github.io/emulators/
+Project-URL: Repository, https://github.com/pasqal-io/emulators
+Project-URL: Issues, https://github.com/pasqal-io/emulators/issues
 Author-email: Anton Quelle <anton.quelle@pasqal.com>, Mauro Mendizabal <mauro.mendizabal-pico@pasqal.com>, Stefano Grava <stefano.grava@pasqal.com>, Pablo Le Henaff <pablo.le-henaff@pasqal.com>
 License: PASQAL OPEN-SOURCE SOFTWARE LICENSE AGREEMENT (MIT-derived)
         
@@ -28,75 +31,8 @@ Description-Content-Type: text/markdown
 
 <div align="center">
   <img src="docs/logos/LogoTaglineSoftGreen.svg">
-
-  # Emu-MPS
 </div>
 
-**Emu-mps** is a backend for the [Pulser low-level Quantum Programming toolkit](https://pulser.readthedocs.io) that lets you run Quantum Algorithms on a simulated device, using GPU acceleration if available. More in depth, emu-mps is designed to **emu**late the dynamics of programmable arrays of neutral atoms, with matrix product states (**mps**). While benchmarking is incomplete as of this writing, early results suggest that this design makes emu-mps faster and more memory-efficient than previous generations of quantum emulators at running simulations with large numbers of qubits.
-
-As of this writing, Emu-MPS is provided for Linux and macOS but will not work under Windows.
-
-## Installation
-
-**Warning:** installing emu-mps will update pulser-core
-
-### Using `hatch`, `uv` or any pyproject-compatible Python manager
-
-To add `emu-mps` to your project, edit your `pyproject.toml` to add the line
-
-```toml
-  "emu-mps"
-```
-
-to the list of `dependencies`.
-
-
-### Using `pip` or `pipx`
-To install the `pipy` package using `pip` or `pipx`
-
-1. Create a `venv` if that's not done yet
-
-```sh
-$ python -m venv venv
-
-```
-
-2. Enter the venv
-
-If you're running Unix:
-
-```sh
-$ . venv/bin/activate
-```
-
-If you're running Windows:
-
-```sh
-C:\> /path/to/new/virtual/environment/Scripts/activate
-```
-
-3. Install the package
-
-```sh
-$ pip install emu-mps
-# or
-$ pipx install emu-mps
-```
-
-
-Join us on [Slack](https://pasqalworkspace.slack.com/archives/C07MUV5K7EU) or by [e-mail](mailto:emulation@pasqal.com) to give us feedback about how you plan to use Emu-MPS or if you require specific feature-upgrades.
-
-## Usage
-
-For the time being, the easiest way to learn how to use this package is to look
-at the [examples](examples/emu_mps_examples) and [notebooks](https://pasqal-io.github.io/emulators/latest/).
-
-See also the [full documentation](https://github.com/pasqal-io/emulators/blob/main/docs/index.md) for
-the API, information about contributing, benchmarks, etc.
-
-
-## Getting in touch
+# Welcome to the Pasqal analog emulators
 
-- [Pasqal Community Portal](https://community.pasqal.com/) (forums, chat, tutorials, examples, code library).
-- [GitHub Repository](https://github.com/pasqal-io/quantum-evolution-kernel) (source code, issue tracker).
-- [Professional Support](https://www.pasqal.com/contact-us/) (if you need tech support, custom licenses, a variant of this library optimized for your workload, your own QPU, remote access to a QPU, ...)
+Welcome, and please see the [GitHub pages](https://pasqal-io.github.io/emulators/) for a landing page to this repo.

{emu_base-1.2.5.dist-info → emu_base-1.2.6.dist-info}/RECORD

@@ -1,6 +1,7 @@
-emu_base/__init__.py,sha256=5X57qf5fo0zv8vmeaDbrW9ThsAEIjy5HPsCpQEIL8iI,1130
+emu_base/__init__.py,sha256=YDB_UjuJJoFfzL3gG_z09ieMTU7RvE5UAGn7POqiYbE,1186
+emu_base/constants.py,sha256=41LYkKLUCz-oxPbd-j7nUDZuhIbUrnez6prT0uR0jcE,56
 emu_base/lindblad_operators.py,sha256=eXkXsLt_fV7jhF31EkxzYFU04rOJV3uWXsl6t1aTAqs,1562
-emu_base/pulser_adapter.py,sha256=vnIiOpjJBXEwMmuWqOdRIlHmkkYnvvPmsSfRNPzBi2k,9118
+emu_base/pulser_adapter.py,sha256=Doi24xNQp09UhyrvNkS_ydXiRV4XNinoziALo6Gil2U,10456
 emu_base/utils.py,sha256=RM8O0qfPAJfcdqqAojwEEKV7I3ZfVDklnTisTGhUg5k,233
 emu_base/base_classes/__init__.py,sha256=Su6fHtjCyg0fw-7y7e7nbMfDASppNRQs8iGaAOkO3c4,570
 emu_base/base_classes/aggregators.py,sha256=bcvoGfZCkPKv-CI29gTma6HBphGh7bjBq2Ome77eRJM,1840
@@ -14,6 +15,6 @@ emu_base/base_classes/state.py,sha256=7iIyZmBqqJ6G4SyYZ3kyylWjAqiYIx0aW5B0T74EPZ
 emu_base/math/__init__.py,sha256=6BbIytYV5uC-e5jLMtIErkcUl_PvfSNnhmVFY9Il8uQ,97
 emu_base/math/brents_root_finding.py,sha256=AVx6L1Il6rpPJWrLJ7cn6oNmJyZOPRgEaaZaubC9lsU,3711
 emu_base/math/krylov_exp.py,sha256=UCFNeq-j2ukgBsOPC9_Jiv1aqpy88SrslDLiCxIGBwk,3840
-emu_base-1.2.5.dist-info/METADATA,sha256=wD4pmz4VvisVEZmM3Z44CC_fiwB6SMw-O9fFKPHvb7w,5576
-emu_base-1.2.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_base-1.2.5.dist-info/RECORD,,
+emu_base-1.2.6.dist-info/METADATA,sha256=AKKbxP9g8y0vjIYHix98y-DxKOlAufhKRW8YQy74h_Q,3522
+emu_base-1.2.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_base-1.2.6.dist-info/RECORD,,