Merge branch 'release25.06public' into 'main'

Public update of release version 25.06 to github

See merge request cuda-hpc-libraries/cuquantum-sdk/cuquantum-public!34

Author: daniellowell

python/README.md

@@ -14,16 +14,15 @@ For instructions on installing *cuQuantum Python*, refer to our
 The build-time dependencies of the cuQuantum Python package include:
 
 * CUDA Toolkit 11.x or 12.x
-* cuStateVec 1.8.0+
-* cuTensorNet 2.7.0+
-* cuDensityMat >=0.1.0
-* Python 3.10+
-* Cython >=0.29.22,<3
+* Python >=3.11, <3.14
+* Cython >=0.29.22
 * pip 21.3.1+
 * [packaging](https://packaging.pypa.io/en/latest/)
 * setuptools 61.0.0+
 * wheel 0.34.0+
 
+> **Note:** Starting with cuQuantum Python v25.06, cuQuantum C libraries including cuDensityMat, cuStateVec and cuTensorNet are no longer build-time dependencies. However, they are still required at runtime.
+
 Except for CUDA and Python, the rest of the build-time dependencies are handled by the new PEP-517-based build system (see Step 7 below).
 
 To compile and install cuQuantum Python from source, please follow the steps below:
@@ -56,14 +55,14 @@ Runtime dependencies of the cuQuantum Python package include:
 * An NVIDIA GPU with compute capability 7.0+
 * Driver: Linux (450.80.02+ for CUDA 11, 525.60.13+ for CUDA 12)
 * CUDA Toolkit 11.x or 12.x
-* cuStateVec 1.8.0+
-* cuTensorNet 2.6.0+
-* cuDensityMat >=0.0.5, <0.1.0
-* Python 3.10+
+* cuStateVec 1.9.0+
+* cuTensorNet 2.8.0+
+* cuDensityMat >=0.2.0, <0.3.0
+* Python >=3.11, <3.14
 * NumPy v1.21+
 * CuPy v13.0.0+ (see [installation guide](https://docs.cupy.dev/en/stable/install.html))
 * PyTorch v1.10+ (optional, see [installation guide](https://pytorch.org/get-started/locally/))
-* Qiskit v0.24.0+ (optional, see [installation guide](https://qiskit.org/documentation/getting_started.html))
+* Qiskit v1.4.2+ (optional, see [installation guide](https://qiskit.org/documentation/getting_started.html))
 * Cirq v0.6.0+ (optional, see [installation guide](https://quantumai.google/cirq/install))
 * mpi4py v3.1.0+ (optional, see [installation guide](https://mpi4py.readthedocs.io/en/stable/install.html))
 
@@ -77,7 +76,6 @@ to your `LD_LIBRARY_PATH` environment variable, and that a compatible CuPy is in
 
 Known issues:
 - If a system has multiple copies of cuTENSOR, one of which is installed in a default system path, the Python runtime could pick it up despite cuQuantum Python is linked to another copy installed elsewhere, potentially causing a version-mismatch error. The proper fix is to remove cuTENSOR from the system paths to ensure the visibility of the proper copy. **DO NOT ATTEMPT** to use `LD_PRELOAD` to overwrite it --- it could cause hard to debug behaviors!
-- In certain environments, if PyTorch is installed `import cuquantum` could fail (with a segmentation fault). It is currently under investigation and a temporary workaround is to import `torch` before importing `cuquantum`.
 - Please ensure that you use consistent binaries and packages for either CUDA 11 or 12. Mixing-and-matching will result in undefined behavior.
 
 ### Samples

python/builder/pep517.py

@@ -6,7 +6,9 @@
 import os
 import site
 import sys
+import logging
 
+logger = logging.getLogger(__name__)
 
 def get_lib_path(name):
     """Get the loaded shared library path."""
@@ -54,11 +56,21 @@ def get_lib_path(name):
     return lib.pop()
 
 
+def _get_cuquantum_lib(lib):
+    try:
+        path = os.path.normpath(get_lib_path(f"lib{lib}.so"))
+        return path
+    except Exception as e:
+        logger.warning(f"library {lib} not found: {e}")
+        return None
+
+
 def _get_cuquantum_libs():
     paths = set()
     for lib in ('custatevec', 'cutensornet', 'cutensor', 'cudensitymat'):
-        path = os.path.normpath(get_lib_path(f"lib{lib}.so"))
-        paths.add(path)
+        path = _get_cuquantum_lib(lib)
+        if path is not None:
+            paths.add(path)
     return tuple(paths)
 
 
@@ -70,21 +82,20 @@ def _get_cuquantum_includes():
             path = os.path.normpath(os.path.join(path, '../include'))
         else:
             path = os.path.join(path, 'include')
-        assert os.path.isdir(path), f"path={path} is invalid"
-        paths.add(path)
+
+        if os.path.isdir(path):
+            paths.add(path)
     return tuple(paths)
 
 
 def _get_cuquantum_target(target):
-    target = f"lib{target}.so"
-    libs = [os.path.basename(lib) for lib in _get_cuquantum_libs()]
-    for lib in libs:
-        if target in lib:
-            lib = '.'.join(lib.split('.')[:3])  # keep SONAME
-            flag = f"-l:{lib} "
-            break
-    else:
-        assert False
+    path = _get_cuquantum_lib(target)
+    if path is None:
+        # Return error as the target library was explicitly asked but not found
+        raise RuntimeError(f"library {target} not found")
+    lib = os.path.basename(path)
+    lib = '.'.join(lib.split('.')[:3])  # keep SONAME
+    flag = f"-l:{lib} "
     return flag
 
 

python/cuquantum/__main__.py

@@ -27,7 +27,7 @@
     import torch
     from .tensor_ifc_torch import TorchTensor
     _TENSOR_TYPES['torch']  = TorchTensor
-    torch_asarray = functools.partial(torch.as_tensor, device='cuda')
+    torch_asarray = functools.partial(torch.tensor, device='cuda')
 except ImportError as e:
     torch = None
     torch_asarray = None

python/cuquantum/_internal/tensor_wrapper.py

@@ -492,21 +492,21 @@ def get_mpi_comm_pointer(comm):
     mpi_comm_size = MPI._sizeof(MPI.Comm)
     return comm_ptr, mpi_comm_size
 
-def deprecate_function(my_func, message):
+def deprecate_function(my_func, message, deprecation_class):
     def add_deprecation_warning(message):
         def decorator(func):
             @functools.wraps(func) 
             def wrapper(*args, **kwargs):
-                warnings.warn(message, DeprecationWarning, stacklevel=2)
+                warnings.warn(message, deprecation_class, stacklevel=2)
                 return func(*args, **kwargs)
             return wrapper
         return decorator
     return add_deprecation_warning(message)(my_func)
 
-def deprecate_class(cls, message):
+def deprecate_class(cls, message, deprecation_class):
     class DeprecatedClass(cls):
         def __new__(cls, *args, **kwargs):
-            warnings.warn(message, DeprecationWarning, stacklevel=2)
+            warnings.warn(message, deprecation_class, stacklevel=2)
             return super(DeprecatedClass, cls).__new__(cls)
 
         def __init__(self, *args, **kwargs):
@@ -516,10 +516,10 @@ def __init__(self, *args, **kwargs):
     DeprecatedClass.__doc__ = cls.__doc__
     return DeprecatedClass
 
-def deprecate(api, message):
+def deprecate(api, message, deprecation_class):
     if inspect.isfunction(api):
-        return deprecate_function(api, message)
+        return deprecate_function(api, message, deprecation_class)
     elif inspect.isclass(api):
-        return deprecate_class(api, message)
+        return deprecate_class(api, message, deprecation_class)
     else:
         raise ValueError(f'API type {type(api)} not supported')

python/cuquantum/_internal/utils.py

@@ -5,4 +5,4 @@
 # Note: cuQuantum Python follows the cuQuantum SDK version, which is now
 # switched to YY.MM and is different from individual libraries' (semantic)
 # versioning scheme.
-__version__ = '25.03.0'
+__version__ = '25.06.0'

python/cuquantum/_version.py

@@ -1,5 +1,13 @@
-# Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES
+# Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES
 #
 # SPDX-License-Identifier: BSD-3-Clause
 
-from cuquantum.bindings import cudensitymat
+from cuquantum.bindings import cudensitymat
+from cuquantum.bindings import custatevec
+from cuquantum.bindings import cutensornet
+
+__all__ = [
+    "cudensitymat",
+    "custatevec",
+    "cutensornet",
+]