cuquantum-python 25.6.0__tar.gz → 25.9.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cuquantum-python
3
- Version: 25.6.0
3
+ Version: 25.9.1
4
4
  Summary: NVIDIA cuQuantum Python
5
5
  Home-page: https://developer.nvidia.com/cuquantum-sdk
6
6
  Author: NVIDIA Corporation
@@ -13,10 +13,12 @@ Project-URL: Source Code, https://github.com/NVIDIA/cuQuantum
13
13
  Keywords: cuda,nvidia,state vector,tensor network,high-performance computing,quantum computing
14
14
  Classifier: Topic :: Scientific/Engineering
15
15
  Classifier: Environment :: GPU :: NVIDIA CUDA
16
- Classifier: Environment :: GPU :: NVIDIA CUDA :: 11
17
16
  Classifier: Environment :: GPU :: NVIDIA CUDA :: 12
17
+ Classifier: Environment :: GPU :: NVIDIA CUDA :: 13
18
18
  Description-Content-Type: text/x-rst
19
19
  License-File: LICENSE
20
+ Provides-Extra: jax
21
+ Requires-Dist: cuquantum-python-jax; extra == "jax"
20
22
  Dynamic: author
21
23
  Dynamic: author-email
22
24
  Dynamic: classifier
@@ -27,6 +29,7 @@ Dynamic: keywords
27
29
  Dynamic: license
28
30
  Dynamic: license-file
29
31
  Dynamic: project-url
32
+ Dynamic: provides-extra
30
33
  Dynamic: summary
31
34
 
32
35
  *****************************************************************************************************
@@ -51,17 +54,14 @@ Installation
51
54
  .. note::
52
55
 
53
56
  Starting cuQuantum 22.11, this package is a meta package pointing to ``cuquantum-python-cuXX``,
54
- where XX is the CUDA major version (currently CUDA 11 & 12 are supported).
55
- The meta package will attempt to infer and install the correct ``-cuXX`` wheel. However,
56
- in situations where the auto-detection fails, this package currently points to ``cuquantum-python-cu11``
57
- with a warning raised (if the verbosity flag ``-v`` is set, as shown above). This behavior
58
- will change in the next release, moving from cu11 to cu12, and users are encouraged to install the new wheels that
57
+ where XX is the CUDA major version (currently CUDA 12 & 13 are supported).
58
+ The meta package will attempt to infer and install the correct ``-cuXX`` wheel.
59
+ The auto-detection mechanism is not guaranteed to work in certain environments, and users are encouraged to install the new wheels that
59
60
  come *with* the ``-cuXX`` suffix.
60
61
 
61
62
  The argument ``--no-cache-dir`` is required for pip 23.1+. It forces pip to execute the
62
63
  auto-detection logic.
63
64
 
64
- Future support for CUDA 11 will be deprecated when support for CUDA 13 is added.
65
65
 
66
66
  Citing cuQuantum
67
67
  ================
@@ -20,17 +20,14 @@ Installation
20
20
  .. note::
21
21
 
22
22
  Starting cuQuantum 22.11, this package is a meta package pointing to ``cuquantum-python-cuXX``,
23
- where XX is the CUDA major version (currently CUDA 11 & 12 are supported).
24
- The meta package will attempt to infer and install the correct ``-cuXX`` wheel. However,
25
- in situations where the auto-detection fails, this package currently points to ``cuquantum-python-cu11``
26
- with a warning raised (if the verbosity flag ``-v`` is set, as shown above). This behavior
27
- will change in the next release, moving from cu11 to cu12, and users are encouraged to install the new wheels that
23
+ where XX is the CUDA major version (currently CUDA 12 & 13 are supported).
24
+ The meta package will attempt to infer and install the correct ``-cuXX`` wheel.
25
+ The auto-detection mechanism is not guaranteed to work in certain environments, and users are encouraged to install the new wheels that
28
26
  come *with* the ``-cuXX`` suffix.
29
27
 
30
28
  The argument ``--no-cache-dir`` is required for pip 23.1+. It forces pip to execute the
31
29
  auto-detection logic.
32
30
 
33
- Future support for CUDA 11 will be deprecated when support for CUDA 13 is added.
34
31
 
35
32
  Citing cuQuantum
36
33
  ================
@@ -124,6 +124,7 @@ def _get_cuda_version() -> Optional[int]:
124
124
 
125
125
  # First try NVRTC
126
126
  libnames = [
127
+ 'libnvrtc.so.13',
127
128
  'libnvrtc.so.12',
128
129
  'libnvrtc.so.11.2',
129
130
  'libnvrtc.so.11.1',
@@ -139,6 +140,7 @@ def _get_cuda_version() -> Optional[int]:
139
140
 
140
141
  # Next try CUDART
141
142
  libnames = [
143
+ 'libcudart.so.13',
142
144
  'libcudart.so.12',
143
145
  'libcudart.so.11.0', # side-effect: a CUDA context would be initialized
144
146
  ]
@@ -176,10 +178,13 @@ def _cuda_version_to_package(ver: int) -> str:
176
178
  f'Your CUDA version ({ver}) is too old.')
177
179
  elif ver < 12000:
178
180
  # CUDA 11.x
179
- suffix = '11'
181
+ raise RuntimeError(f'CUDA 11.x is not supported. Please install CUDA 12.x or later.')
180
182
  elif ver < 13000:
181
183
  # CUDA 12.x
182
184
  suffix = '12'
185
+ elif ver < 14000:
186
+ # CUDA 13.x
187
+ suffix = '13'
183
188
  else:
184
189
  raise AutoDetectionFailed(
185
190
  f'Your CUDA version ({ver}) is too new.')
@@ -190,7 +195,7 @@ def _cuda_version_to_package(ver: int) -> str:
190
195
 
191
196
  # "Public" API to the caller
192
197
  def infer_best_package(package_name: str,
193
- package_supported_cuda_ver: List[str] = ['11', '12']) -> str:
198
+ package_supported_cuda_ver: List[str] = ['12', '13']) -> str:
194
199
  """Returns the appropriate wheel name for the environment."""
195
200
 
196
201
  global PACKAGE_NAME, PACKAGE_SUPPORTED_CUDA_VER
@@ -218,7 +223,7 @@ def infer_best_package(package_name: str,
218
223
  )
219
224
  raise AutoDetectionFailed(message)
220
225
 
221
- # Disallow -cu11 & -cu12 wheels from coexisting
226
+ # Disallow -cu12 & -cu13 wheels from coexisting
222
227
  if len(installed) > 1 or (len(installed) == 1 and installed[0] != to_install):
223
228
  raise AutoDetectionFailed(
224
229
  f'You already have the {PACKAGE_NAME} package(s) installed: \n'
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cuquantum-python
3
- Version: 25.6.0
3
+ Version: 25.9.1
4
4
  Summary: NVIDIA cuQuantum Python
5
5
  Home-page: https://developer.nvidia.com/cuquantum-sdk
6
6
  Author: NVIDIA Corporation
@@ -13,10 +13,12 @@ Project-URL: Source Code, https://github.com/NVIDIA/cuQuantum
13
13
  Keywords: cuda,nvidia,state vector,tensor network,high-performance computing,quantum computing
14
14
  Classifier: Topic :: Scientific/Engineering
15
15
  Classifier: Environment :: GPU :: NVIDIA CUDA
16
- Classifier: Environment :: GPU :: NVIDIA CUDA :: 11
17
16
  Classifier: Environment :: GPU :: NVIDIA CUDA :: 12
17
+ Classifier: Environment :: GPU :: NVIDIA CUDA :: 13
18
18
  Description-Content-Type: text/x-rst
19
19
  License-File: LICENSE
20
+ Provides-Extra: jax
21
+ Requires-Dist: cuquantum-python-jax; extra == "jax"
20
22
  Dynamic: author
21
23
  Dynamic: author-email
22
24
  Dynamic: classifier
@@ -27,6 +29,7 @@ Dynamic: keywords
27
29
  Dynamic: license
28
30
  Dynamic: license-file
29
31
  Dynamic: project-url
32
+ Dynamic: provides-extra
30
33
  Dynamic: summary
31
34
 
32
35
  *****************************************************************************************************
@@ -51,17 +54,14 @@ Installation
51
54
  .. note::
52
55
 
53
56
  Starting cuQuantum 22.11, this package is a meta package pointing to ``cuquantum-python-cuXX``,
54
- where XX is the CUDA major version (currently CUDA 11 & 12 are supported).
55
- The meta package will attempt to infer and install the correct ``-cuXX`` wheel. However,
56
- in situations where the auto-detection fails, this package currently points to ``cuquantum-python-cu11``
57
- with a warning raised (if the verbosity flag ``-v`` is set, as shown above). This behavior
58
- will change in the next release, moving from cu11 to cu12, and users are encouraged to install the new wheels that
57
+ where XX is the CUDA major version (currently CUDA 12 & 13 are supported).
58
+ The meta package will attempt to infer and install the correct ``-cuXX`` wheel.
59
+ The auto-detection mechanism is not guaranteed to work in certain environments, and users are encouraged to install the new wheels that
59
60
  come *with* the ``-cuXX`` suffix.
60
61
 
61
62
  The argument ``--no-cache-dir`` is required for pip 23.1+. It forces pip to execute the
62
63
  auto-detection logic.
63
64
 
64
- Future support for CUDA 11 will be deprecated when support for CUDA 13 is added.
65
65
 
66
66
  Citing cuQuantum
67
67
  ================
@@ -6,4 +6,5 @@ cuquantum_python.egg-info/PKG-INFO
6
6
  cuquantum_python.egg-info/SOURCES.txt
7
7
  cuquantum_python.egg-info/dependency_links.txt
8
8
  cuquantum_python.egg-info/not-zip-safe
9
+ cuquantum_python.egg-info/requires.txt
9
10
  cuquantum_python.egg-info/top_level.txt
@@ -0,0 +1,3 @@
1
+
2
+ [jax]
3
+ cuquantum-python-jax
@@ -1,4 +1,4 @@
1
- # Copyright (c) 2021-2024, NVIDIA CORPORATION & AFFILIATES.
1
+ # Copyright (c) 2021-2025, NVIDIA CORPORATION & AFFILIATES.
2
2
  #
3
3
  # SPDX-License-Identifier: BSD-3-Clause
4
4
 
@@ -17,7 +17,7 @@ from cuda_autodetect import infer_best_package, bdist_wheel
17
17
 
18
18
  # Update this for every release
19
19
  # Must pin to the cuquantum-python-cuXX version
20
- package_ver = "25.06.0"
20
+ package_ver = "25.09.1"
21
21
  package_name = "cuquantum-python"
22
22
 
23
23
 
@@ -41,15 +41,14 @@ else:
41
41
  install_requires = [f"{infer_best_package(package_name)}=={package_ver}",]
42
42
  # For cuQuantum 24.03+, we need cuTENSOR 2.0+, meaning we can't use CuPy older than
43
43
  # v13.0.0, and CuPy v13+ supports CUDA 11.2+
44
- if 11020 <= cuda_autodetect.CUDA_RESOLUTION < 12000:
45
- # no ambiguity for CUDA 11, cuquantum-python-cu11 would do the right thing
46
- pass
47
- elif 12000 <= cuda_autodetect.CUDA_RESOLUTION < 13000:
44
+ if 12000 <= cuda_autodetect.CUDA_RESOLUTION < 13000:
48
45
  # no ambiguity for CUDA 12, cuquantum-python-cu12 would do the right thing
49
46
  pass
47
+ elif 13000 <= cuda_autodetect.CUDA_RESOLUTION < 14000:
48
+ # no ambiguity for CUDA 13, cuquantum-python-cu13 would do the right thing
49
+ pass
50
50
  else:
51
- # most likely we failed to detect the CUDA version, assume CUDA 11.2+
52
- install_requires.append('cupy-cuda11x')
51
+ raise ValueError(f"Unsupported CUDA version: {cuda_autodetect.CUDA_RESOLUTION}")
53
52
  data_files = []
54
53
  cmdclass = {'bdist_wheel': bdist_wheel} if bdist_wheel is not None else {}
55
54
 
@@ -77,6 +76,9 @@ setup(
77
76
  include_package_data=True,
78
77
  zip_safe=False,
79
78
  data_files=data_files,
79
+ extras_require={
80
+ "jax": ["cuquantum-python-jax"]
81
+ },
80
82
  setup_requires=[
81
83
  "setuptools",
82
84
  "wheel",
@@ -85,8 +87,8 @@ setup(
85
87
  classifiers=[
86
88
  "Topic :: Scientific/Engineering",
87
89
  "Environment :: GPU :: NVIDIA CUDA",
88
- "Environment :: GPU :: NVIDIA CUDA :: 11",
89
90
  "Environment :: GPU :: NVIDIA CUDA :: 12",
91
+ "Environment :: GPU :: NVIDIA CUDA :: 13",
90
92
  ],
91
93
  cmdclass=cmdclass,
92
94
  )