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

sync with internal cuQuantum repo main branch for release (commit 0091f609c)

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

Author: daniellowell

LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2021-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+Copyright (c) 2021-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 
 BSD-3-Clause
 

python/LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2021-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+Copyright (c) 2021-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 
 BSD-3-Clause
 

python/README.md

@@ -14,9 +14,9 @@ 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.7.0+
-* cuTensorNet 2.6.0+
-* cuDensityMat >=0.0.5, <0.1.0
+* cuStateVec 1.8.0+
+* cuTensorNet 2.7.0+
+* cuDensityMat >=0.1.0
 * Python 3.10+
 * Cython >=0.29.22,<3
 * pip 21.3.1+
@@ -56,7 +56,7 @@ 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.7.0+
+* cuStateVec 1.8.0+
 * cuTensorNet 2.6.0+
 * cuDensityMat >=0.0.5, <0.1.0
 * Python 3.10+

python/builder/pep517.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2021-2024, NVIDIA CORPORATION & AFFILIATES
+# Copyright (c) 2021-2025, NVIDIA CORPORATION & AFFILIATES
 #
 # SPDX-License-Identifier: BSD-3-Clause
 
@@ -30,9 +30,9 @@ def get_requires_for_build_wheel(config_settings=None):
     # set up version constraints: note that CalVer like 22.03 is normalized to
     # 22.3 by setuptools, so we must follow the same practice in the constraints;
     # also, we don't need the patch number here
-    cuqnt_require = [f'custatevec-cu{utils.cuda_major_ver}~=1.7',   # ">=1.7.0,<2"
-                     f'cutensornet-cu{utils.cuda_major_ver}~=2.6',   # ">=2.6.0,<3"
-                     f'cudensitymat-cu{utils.cuda_major_ver}~=0.0.5'  # ">=0.0.5, <0.1"
+    cuqnt_require = [f'custatevec-cu{utils.cuda_major_ver}~=1.8',   # ">=1.8.0,<2"
+                     f'cutensornet-cu{utils.cuda_major_ver}~=2.7',   # ">=2.7.0,<3"
+                     f'cudensitymat-cu{utils.cuda_major_ver}~=0.1'  # ">=0.1, <0.2"
                     ]
 
     return _build_meta.get_requires_for_build_wheel(config_settings) + cuqnt_require

python/cuquantum/__init__.py

@@ -1,16 +1,18 @@
-# Copyright (c) 2021-2024, NVIDIA CORPORATION & AFFILIATES
+# Copyright (c) 2021-2025, NVIDIA CORPORATION & AFFILIATES
 #
 # SPDX-License-Identifier: BSD-3-Clause
 
 from cuquantum import bindings
 from cuquantum import custatevec
 from cuquantum import cutensornet
 from cuquantum import densitymat
+from cuquantum import tensornet
 from cuquantum.cutensornet import (
     contract, contract_path, einsum, einsum_path, tensor, tensor_qualifiers_dtype, BaseCUDAMemoryManager, CircuitToEinsum, MemoryPointer, 
     Network, NetworkOptions, OptimizerInfo, OptimizerOptions, PathFinderOptions, 
     ReconfigOptions, SlicerOptions, MemoryLimitExceeded)
-from cuquantum._utils import ComputeType, cudaDataType, libraryPropertyType
+from cuquantum.bindings._utils import ComputeType, cudaDataType, libraryPropertyType
+from cuquantum._internal import enum_utils
 from cuquantum._version import __version__
 
 
@@ -27,23 +29,35 @@
         custatevec.CommunicatorType,
         custatevec.DataTransferType,
         custatevec.StateVectorType,
+        cutensornet.Status,
         cutensornet.ContractionOptimizerInfoAttribute,
         cutensornet.ContractionOptimizerConfigAttribute,
         cutensornet.ContractionAutotunePreferenceAttribute,
+        cutensornet.NetworkAttribute,
         cutensornet.WorksizePref,
+        cutensornet.WorkspaceKind,
         cutensornet.Memspace,
+        cutensornet.SmartOption,
         cutensornet.GraphAlgo,
         cutensornet.MemoryModel,
         cutensornet.OptimizerCost,
         cutensornet.TensorSVDConfigAttribute,
+        cutensornet.TensorSVDAlgo,
         cutensornet.TensorSVDNormalization,
         cutensornet.TensorSVDPartition,
         cutensornet.TensorSVDInfoAttribute,
         cutensornet.GateSplitAlgo,
+        cutensornet.BoundaryCondition,
         cutensornet.StatePurity,
+        cutensornet.StateAttribute,
         cutensornet.MarginalAttribute,
         cutensornet.SamplerAttribute,
+        cutensornet.AccessorAttribute,
+        cutensornet.ExpectationAttribute,
+        cutensornet.StateMPOApplication,
+        cutensornet.StateMPSGaugeOption,
         ):
-    cutensornet._internal.enum_utils.add_enum_class_doc(enum, chomp="_ATTRIBUTE|_PREFERENCE_ATTRIBUTE")
+    enum_utils.add_enum_class_doc(enum, chomp="_ATTRIBUTE|_PREFERENCE_ATTRIBUTE")
 
 del enum
+del enum_utils

python/cuquantum/__main__.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2021-2024, NVIDIA CORPORATION & AFFILIATES
+# Copyright (c) 2021-2025, NVIDIA CORPORATION & AFFILIATES
 #
 # SPDX-License-Identifier: BSD-3-Clause
 
@@ -16,11 +16,11 @@ def get_lib_path(name):
 
     # We have switched to use dlopen, force library loading via internal API
     if "custatevec" in name:
-        from cuquantum import custatevec as cusv
-        cusv._internal.custatevec._inspect_function_pointers()
+        from cuquantum import bindings
+        bindings._internal.custatevec._inspect_function_pointers()
     elif "cutensor" in name:  # cutensor or cutensornet
-        from cuquantum import cutensornet as cutn
-        cutn._internal.cutensornet._inspect_function_pointers()
+        from cuquantum import bindings
+        bindings._internal.cutensornet._inspect_function_pointers()
     elif "cudensitymat" in name:
         from cuquantum import bindings
         bindings._internal.cudensitymat._inspect_function_pointers()

python/cuquantum/_internal/__init__.py

@@ -0,0 +1,3 @@
+# Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES
+#
+# SPDX-License-Identifier: BSD-3-Clause