pennylane-lightning-tensor 0.42.0__cp311-cp311-manylinux_2_28_aarch64.whl → 0.43.0__cp311-cp311-manylinux_2_28_aarch64.whl

pennylane_lightning/lightning_tensor/_measurements.py

@@ -34,7 +34,6 @@ import pennylane as qml
 from pennylane.devices.qubit.sampling import _group_measurements
 from pennylane.measurements import (
     ClassicalShadowMP,
-    CountsMP,
     ExpectationMP,
     MeasurementProcess,
     ProbabilityMP,
@@ -373,15 +372,7 @@ class LightningTensorMeasurements:
         wires = reduce(sum, (mp.wires for mp in mps))
 
         def _process_single_shot(samples):
-            processed = []
-            for mp in mps:
-                res = mp.process_samples(samples, wires)
-                if not isinstance(mp, CountsMP):
-                    res = qml.math.squeeze(res)
-
-                processed.append(res)
-
-            return tuple(processed)
+            return tuple(mp.process_samples(samples, wires) for mp in mps)
 
         try:
             samples = self._measurement_lightning.generate_samples(

pennylane_lightning/lightning_tensor/_tensornet.py

@@ -405,6 +405,7 @@ class LightningTensorNet:
         cutoff_mode (str): Singular value truncation mode for MPS tensors can be done either by
             considering the absolute values of the singular values (``"abs"``) or by considering
             the relative values of the singular values (``"rel"``). Default is ``"abs"``.
+        worksize_pref (str): Preference for workspace size for cutensornet backend. The options are ``recommended``, ``min``, and ``max``. Default is ``recommended``.
     """
 
     # pylint: disable=too-many-arguments, too-many-positional-arguments
@@ -439,6 +440,9 @@ class LightningTensorNet:
         else:
             raise DeviceError(f"The method {self._method} is not supported.")
 
+        worksize_pref = kwargs.get("worksize_pref", "recommended")
+        self.set_worksize_pref(worksize_pref)
+
     @property
     def dtype(self):
         """Returns the tensor network data type."""
@@ -648,6 +652,8 @@ class LightningTensorNet:
             gate_matrix, wires, max_mpo_bond_dim, self._c_dtype
         )
 
+        # Convert to C-contiguous arrays for C++ bindings
+        mpos = [np.ascontiguousarray(mpo) for mpo in mpos]
         self._tensornet.applyMPOOperation(mpos, sorted_wires, max_mpo_bond_dim)
 
     # pylint: disable=too-many-branches
@@ -765,6 +771,18 @@ class LightningTensorNet:
 
         self._apply_lightning(operations)
 
+    def set_worksize_pref(self, worksize_pref: str):
+        """Set the worksize preference for the cutensornet backend.
+
+        Args:
+            worksize_pref (str): Preference for workspace size for cutensornet backend. The options are ``recommended``, ``max``, or ``min``. Default is ``recommended``.
+        """
+        if worksize_pref not in ("recommended", "max", "min"):
+            raise ValueError(
+                f'Worksize preference "{worksize_pref}" is not valid. Please select one of the following options: "recommended", "max", or "min".'
+            )
+        self._tensornet.setWorksizePref(worksize_pref)
+
     def set_tensor_network(self, circuit: QuantumScript):
         """
         Set the tensor network that results from executing the given quantum script.

pennylane_lightning/lightning_tensor/lightning_tensor.py

@@ -17,15 +17,16 @@ It is a device to perform tensor network simulations of quantum circuits using `
 """
 from dataclasses import replace
 from numbers import Number
-from typing import Callable, Optional, Sequence, Tuple, Union
+from typing import Callable, Optional, Sequence, Tuple
 from warnings import warn
 
 import numpy as np
 import pennylane as qml
-from pennylane.devices import DefaultExecutionConfig, Device, ExecutionConfig
+from pennylane.devices import Device, ExecutionConfig
 from pennylane.devices.modifiers import simulator_tracking, single_tape_support
 from pennylane.devices.preprocess import (
     decompose,
+    device_resolve_dynamic_wires,
     validate_device_wires,
     validate_measurements,
     validate_observables,
@@ -57,9 +58,9 @@ except ImportError as ex:
     warn(str(ex), UserWarning)
     LT_CPP_BINARY_AVAILABLE = False
 
-Result_or_ResultBatch = Union[Result, ResultBatch]
+Result_or_ResultBatch = Result | ResultBatch
 QuantumTapeBatch = Sequence[QuantumTape]
-QuantumTape_or_Batch = Union[QuantumTape, QuantumTapeBatch]
+QuantumTape_or_Batch = QuantumTape | QuantumTapeBatch
 PostprocessingFn = Callable[[ResultBatch], Result_or_ResultBatch]
 
 
@@ -72,7 +73,6 @@ _methods = frozenset({"mps", "tn"})
 _operations = frozenset(
     {
         "Identity",
-        "BasisState",
         "MPSPrep",
         "QubitUnitary",
         "ControlledQubitUnitary",
@@ -175,9 +175,6 @@ def stopping_condition(op: Operator) -> bool:
     if isinstance(op, qml.MPSPrep):
         return True
 
-    if op.name in ("C(SProd)", "C(Exp)"):
-        return True
-
     return op.has_matrix and op.name in _operations
 
 
@@ -245,6 +242,7 @@ class LightningTensor(Device):
         cutoff (float): (Only for ``method=mps``) The threshold used to truncate the singular values of the MPS tensors. The default is 0.
         cutoff_mode (str): (Only for ``method=mps``) Singular value truncation mode for MPS tensors. The options are ``"rel"`` and ``"abs"``. Default is ``"abs"``.
         backend (str): Supported backend. Currently, only ``cutensornet`` is supported. Default is ``cutensornet``.
+        worksize_pref (str): Preference for workspace size for cutensornet backend. The options are ``recommended``, ``min``, and ``max``. Default is ``recommended``.
 
     **Example for the MPS method**
 
@@ -290,12 +288,10 @@ class LightningTensor(Device):
     """
 
     # pylint: disable=too-many-instance-attributes
-    pennylane_requires = ">=0.41"
-    version = __version__
 
     _device_options = {
-        "mps": ("backend", "max_bond_dim", "cutoff", "cutoff_mode"),
-        "tn": ("backend"),
+        "mps": ("backend", "max_bond_dim", "cutoff", "cutoff_mode", "worksize_pref"),
+        "tn": ("backend", "worksize_pref"),
     }
 
     _CPP_BINARY_AVAILABLE = LT_CPP_BINARY_AVAILABLE
@@ -340,6 +336,7 @@ class LightningTensor(Device):
         self._c_dtype = c_dtype
 
         self._backend = kwargs.get("backend", "cutensornet")
+        self._worksize_pref = kwargs.get("worksize_pref", "recommended")
 
         for arg in kwargs:
             if arg not in self._device_options[self._method]:
@@ -397,19 +394,22 @@ class LightningTensor(Device):
                 max_bond_dim=self._max_bond_dim,
                 cutoff=self._cutoff,
                 cutoff_mode=self._cutoff_mode,
+                worksize_pref=self._worksize_pref,
             )
         return LightningTensorNet(num_wires, self._method, self._c_dtype, device_name=self.name)
 
     dtype = c_dtype
 
-    def _setup_execution_config(
-        self, config: Optional[ExecutionConfig] = DefaultExecutionConfig
+    # pylint: disable=unused-argument
+    def setup_execution_config(
+        self, config: ExecutionConfig | None = None, circuit=None
     ) -> ExecutionConfig:
         """
         Update the execution config with choices for how the device should be used and the device options.
         """
+        if config is None:
+            config = ExecutionConfig()
         # TODO: add options for gradients next quarter
-
         updated_values = {}
 
         new_device_options = dict(config.device_options)
@@ -431,10 +431,10 @@ class LightningTensor(Device):
 
         return circuit.map_to_standard_wires() if self.num_wires is None else circuit
 
-    def preprocess(
+    def preprocess_transforms(
         self,
-        execution_config: ExecutionConfig = DefaultExecutionConfig,
-    ):
+        execution_config: ExecutionConfig | None = None,
+    ) -> TransformProgram:
         """This function defines the device transform program to be applied and an updated device configuration.
 
         Args:
@@ -451,28 +451,30 @@ class LightningTensor(Device):
         * Does not support derivatives.
         * Does not support vector-Jacobian products.
         """
-
-        config = self._setup_execution_config(execution_config)
+        if execution_config is None:
+            execution_config = self.setup_execution_config(ExecutionConfig())
 
         program = TransformProgram()
 
         program.add_transform(validate_measurements, name=self.name)
         program.add_transform(validate_observables, accepted_observables, name=self.name)
-        program.add_transform(validate_device_wires, self._wires, name=self.name)
         program.add_transform(
             decompose,
             stopping_condition=stopping_condition,
-            stopping_condition_shots=stopping_condition,
             skip_initial_state_prep=True,
             name=self.name,
+            device_wires=self.wires,
+            target_gates=self.operations,
         )
-        return program, config
+        program.add_transform(device_resolve_dynamic_wires, wires=self.wires, allow_resets=False)
+        program.add_transform(validate_device_wires, self._wires, name=self.name)
+        return program
 
     # pylint: disable=unused-argument
     def execute(
         self,
         circuits: QuantumTape_or_Batch,
-        execution_config: ExecutionConfig = DefaultExecutionConfig,
+        execution_config: ExecutionConfig | None = None,
     ) -> Result_or_ResultBatch:
         """Execute a circuit or a batch of circuits and turn it into results.
 
@@ -503,7 +505,7 @@ class LightningTensor(Device):
     # pylint: disable=unused-argument
     def supports_derivatives(
         self,
-        execution_config: Optional[ExecutionConfig] = None,
+        execution_config: ExecutionConfig | None = None,
         circuit: Optional[qml.tape.QuantumTape] = None,
     ) -> bool:
         """Check whether or not derivatives are available for a given configuration and circuit.
@@ -521,7 +523,7 @@ class LightningTensor(Device):
     def compute_derivatives(
         self,
         circuits: QuantumTape_or_Batch,
-        execution_config: ExecutionConfig = DefaultExecutionConfig,
+        execution_config: ExecutionConfig | None = None,
     ):
         """Calculate the Jacobian of either a single or a batch of circuits on the device.
 
@@ -539,7 +541,7 @@ class LightningTensor(Device):
     def execute_and_compute_derivatives(
         self,
         circuits: QuantumTape_or_Batch,
-        execution_config: ExecutionConfig = DefaultExecutionConfig,
+        execution_config: ExecutionConfig | None = None,
     ):
         """Compute the results and Jacobians of circuits at the same time.
 
@@ -557,7 +559,7 @@ class LightningTensor(Device):
     # pylint: disable=unused-argument
     def supports_vjp(
         self,
-        execution_config: Optional[ExecutionConfig] = None,
+        execution_config: ExecutionConfig | None = None,
         circuit: Optional[QuantumTape] = None,
     ) -> bool:
         """Whether or not this device defines a custom vector-Jacobian product.
@@ -575,7 +577,7 @@ class LightningTensor(Device):
         self,
         circuits: QuantumTape_or_Batch,
         cotangents: Tuple[Number],
-        execution_config: ExecutionConfig = DefaultExecutionConfig,
+        execution_config: ExecutionConfig | None = None,
     ):
         r"""The vector-Jacobian product used in reverse-mode differentiation.
 
@@ -597,7 +599,7 @@ class LightningTensor(Device):
         self,
         circuits: QuantumTape_or_Batch,
         cotangents: Tuple[Number],
-        execution_config: ExecutionConfig = DefaultExecutionConfig,
+        execution_config: ExecutionConfig | None = None,
     ):
         """Calculate both the results and the vector-Jacobian product used in reverse-mode differentiation.
 

{pennylane_lightning_tensor-0.42.0.dist-info → pennylane_lightning_tensor-0.43.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pennylane_lightning_tensor
-Version: 0.42.0
+Version: 0.43.0
 Summary: PennyLane-Lightning plugin
 Maintainer-email: "Xanadu Quantum Technologies Inc." <software@xanadu.ai>
 License-Expression: Apache-2.0
@@ -15,12 +15,11 @@ Classifier: Operating System :: POSIX :: Linux
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Physics
-Requires-Python: >=3.10
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: pennylane>=0.41
@@ -31,7 +30,7 @@ Requires-Dist: nvidia-nvjitlink-cu12
 Requires-Dist: nvidia-cusparse-cu12
 Requires-Dist: nvidia-cublas-cu12
 Requires-Dist: nvidia-cuda-runtime-cu12
-Requires-Dist: pennylane_lightning==0.42.0
+Requires-Dist: pennylane_lightning==0.43.0
 Provides-Extra: gpu
 Requires-Dist: pennylane-lightning-gpu; extra == "gpu"
 Provides-Extra: kokkos
@@ -100,7 +99,7 @@ The Lightning plugin ecosystem provides fast state-vector and tensor-network sim
 
 [PennyLane](https://docs.pennylane.ai) is a cross-platform Python library for quantum machine
 learning, automatic differentiation, and optimization of hybrid quantum-classical computations.
-PennyLane supports Python 3.10 and above.
+PennyLane supports Python 3.11 and above.
 
 ## Backends
 
@@ -213,7 +212,7 @@ enable the device bindings to PennyLane, which are held to their own respective
 
 PennyLane Lightning makes use of the following libraries and tools, which are under their own respective licenses:
 
-- **pybind11:** https://github.com/pybind/pybind11
+- **nanobind:** https://github.com/wjakob/nanobind
 - **Kokkos Core:** https://github.com/kokkos/kokkos
 - **NVIDIA cuQuantum:** https://developer.nvidia.com/cuquantum-sdk
 - **scipy-openblas32:** https://pypi.org/project/scipy-openblas32/

pennylane_lightning_tensor-0.43.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+pennylane_lightning_tensor.libs/libgomp-947d5fa1.so.1.0.0,sha256=3DIx-P1MrcRthlnDuN2BGuMxdHSehoB_FvYerW7lm_0,343345
+pennylane_lightning_tensor-0.43.0.dist-info/WHEEL,sha256=8jTxASeLvPr09PZ22DGSMQlRkrmgeqGaotjla49xWxo,114
+pennylane_lightning_tensor-0.43.0.dist-info/entry_points.txt,sha256=2IpSy50QiwMzPv1qWJeCtp_oEEHsOvaQbRV8G5ExdrM,92
+pennylane_lightning_tensor-0.43.0.dist-info/METADATA,sha256=o7yrYVBdM6HMaR3VPpq_1kjt6SLR0_H51VvVUTdAfaU,11419
+pennylane_lightning_tensor-0.43.0.dist-info/RECORD,,
+pennylane_lightning_tensor-0.43.0.dist-info/top_level.txt,sha256=G8tVqgen55ojHMLJr9_CDbxbX-_frQCZWMCqUvvGIZI,41
+pennylane_lightning_tensor-0.43.0.dist-info/licenses/LICENSE,sha256=_zKpiTfJGLmlVz4pw88ztD4wTe2QkkUNBGSaac54sjM,11720
+pennylane_lightning/lightning_tensor_ops.cpython-311-aarch64-linux-gnu.so,sha256=Hvdqc7earRex_UtbnxQOSVehuAjnzMbSy7RxP0HObAo,1086345
+pennylane_lightning/lightning_tensor/__init__.py,sha256=y2DrEuU_gP2lV2jKVoR7qcpcGO_1LXTqYF6cIS7ULTU,737
+pennylane_lightning/lightning_tensor/lightning_tensor.py,sha256=vpnrYQhXwU-xbmrY2wDqqQduHGSsM3FX5rG2KxOhAt8,22405
+pennylane_lightning/lightning_tensor/_tensornet.py,sha256=q5XL18hItw6xG3y8RhrArCiAEg9M1PJTDJM_DBRxTMU,31546
+pennylane_lightning/lightning_tensor/_measurements.py,sha256=zo3_KPfms_oj-zmeye5j-_i0G7wFuG7boXFS1UxDwZk,16123

pennylane_lightning_tensor-0.42.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-pennylane_lightning_tensor.libs/libgomp-98df74fd.so.1.0.0,sha256=c92itWhSYBOIWRdkLAttSVj5yIr-O0pzI8LXUO4jbGo,343345
-pennylane_lightning_tensor-0.42.0.dist-info/WHEEL,sha256=8jTxASeLvPr09PZ22DGSMQlRkrmgeqGaotjla49xWxo,114
-pennylane_lightning_tensor-0.42.0.dist-info/entry_points.txt,sha256=2IpSy50QiwMzPv1qWJeCtp_oEEHsOvaQbRV8G5ExdrM,92
-pennylane_lightning_tensor-0.42.0.dist-info/METADATA,sha256=6Ztixir6yVNnFy0yYwI7PE-s1N1MALQ60qI9rO-YyQg,11470
-pennylane_lightning_tensor-0.42.0.dist-info/RECORD,,
-pennylane_lightning_tensor-0.42.0.dist-info/top_level.txt,sha256=G8tVqgen55ojHMLJr9_CDbxbX-_frQCZWMCqUvvGIZI,41
-pennylane_lightning_tensor-0.42.0.dist-info/licenses/LICENSE,sha256=_zKpiTfJGLmlVz4pw88ztD4wTe2QkkUNBGSaac54sjM,11720
-pennylane_lightning/lightning_tensor_ops.cpython-311-aarch64-linux-gnu.so,sha256=rdJ2K6LpNrIS2ltvj2YvdJkezmioLOENJHnfMiMruE4,1200145
-pennylane_lightning/lightning_tensor/__init__.py,sha256=y2DrEuU_gP2lV2jKVoR7qcpcGO_1LXTqYF6cIS7ULTU,737
-pennylane_lightning/lightning_tensor/lightning_tensor.py,sha256=ox2YZnFkKz19A0x2lxTOnAUqtHwWDufkpEEvU-_j-Ac,22018
-pennylane_lightning/lightning_tensor/_tensornet.py,sha256=eMc2MdXqqvY1gy0I4qAm1Zizjq7j0ujSBDcNXCb94S0,30520
-pennylane_lightning/lightning_tensor/_measurements.py,sha256=jUsPQq7igJzt9IDlyoMuSo2yMT2QfqGdM3Hos2G4WFQ,16346