From c4d64517562a78bb7818d1ff447cbc9e9e3c99ef Mon Sep 17 00:00:00 2001 From: yangg Date: Wed, 15 Jul 2026 17:08:22 -0700 Subject: [PATCH 1/6] Pathfinder: add support on cutensornet for library loading and header descriptors --- .../cuda/pathfinder/_dynamic_libs/descriptor_catalog.py | 7 +++++++ .../pathfinder/_headers/header_descriptor_catalog.py | 9 +++++++++ cuda_pathfinder/pyproject.toml | 2 ++ cuda_pathfinder/tests/test_find_nvidia_headers.py | 1 + toolshed/conda_create_for_pathfinder_testing.sh | 1 + 5 files changed, 20 insertions(+) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py index de57ff23ad6..36b56c730bd 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py @@ -375,6 +375,13 @@ class DescriptorSpec: site_packages_windows=("cutensor/bin",), dependencies=("cutensor", "cublasLt"), ), + DescriptorSpec( + name="cutensornet", + packaged_with="other", + linux_sonames=("libcutensornet.so.2",), + site_packages_linux=("cuquantum/lib",), + dependencies=("cutensor", "cublas", "cusolver"), + ), DescriptorSpec( name="nccl", packaged_with="other", diff --git a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py index b64fa56cbb4..36aaf65cb93 100644 --- a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py @@ -183,6 +183,15 @@ class HeaderDescriptorSpec: conda_targets_layout=False, use_ctk_root_canary=False, ), + HeaderDescriptorSpec( + name="cutensornet", + packaged_with="other", + header_basename="cutensornet.h", + site_packages_dirs=("cuquantum/include",), + available_on_windows=False, + conda_targets_layout=False, + use_ctk_root_canary=False, + ), HeaderDescriptorSpec( name="cutlass", packaged_with="other", diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index 86f33b726e9..a6621ab5bb9 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -22,6 +22,7 @@ cu12 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*", "cuda-toolkit[cufile]==12.*; sys_platform != 'win32'", "cutensor-cu12", + "cutensornet-cu12; sys_platform != 'win32'", "nvidia-cublasmp-cu12; sys_platform != 'win32'", "nvidia-cudss-cu12", "nvidia-cufftmp-cu12; sys_platform != 'win32'", @@ -35,6 +36,7 @@ cu13 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*", "cuda-toolkit[cufile]==13.*; sys_platform != 'win32'", "cutensor-cu13", + "cutensornet-cu13; sys_platform != 'win32'", "nvidia-cublasmp-cu13; sys_platform != 'win32'", "nvidia-cudla; platform_system == 'Linux' and platform_machine == 'aarch64'", "nvidia-cudss-cu13", diff --git a/cuda_pathfinder/tests/test_find_nvidia_headers.py b/cuda_pathfinder/tests/test_find_nvidia_headers.py index 5296917c35b..dc30feae2b9 100644 --- a/cuda_pathfinder/tests/test_find_nvidia_headers.py +++ b/cuda_pathfinder/tests/test_find_nvidia_headers.py @@ -46,6 +46,7 @@ "cusparseLt": r"^nvidia-cusparselt-.*$", "cute": r"^nvidia-cutlass$", "cutensor": r"^cutensor-.*$", + "cutensornet": r"^cutensornet-.*$", "cutlass": r"^nvidia-cutlass$", "mathdx": r"^nvidia-libmathdx-.*$", "nvshmem": r"^nvidia-nvshmem-.*$", diff --git a/toolshed/conda_create_for_pathfinder_testing.sh b/toolshed/conda_create_for_pathfinder_testing.sh index 69844904921..d6809c1cee9 100755 --- a/toolshed/conda_create_for_pathfinder_testing.sh +++ b/toolshed/conda_create_for_pathfinder_testing.sh @@ -27,6 +27,7 @@ set -u cpkgs=( "cusparselt-dev" "cutensor" + "cutensornet" "cutlass" "libcublasmp-dev" "libcudss-dev" From bde78ab67c603f9fdac32f3f5297e0fbe8bc0e27 Mon Sep 17 00:00:00 2001 From: yangg Date: Wed, 15 Jul 2026 17:20:41 -0700 Subject: [PATCH 2/6] Pathfinder: add support on custatevec for library loading and header descriptors --- .../cuda/pathfinder/_dynamic_libs/descriptor_catalog.py | 7 +++++++ .../pathfinder/_headers/header_descriptor_catalog.py | 9 +++++++++ cuda_pathfinder/pyproject.toml | 2 ++ cuda_pathfinder/tests/test_find_nvidia_headers.py | 3 ++- toolshed/conda_create_for_pathfinder_testing.sh | 1 + 5 files changed, 21 insertions(+), 1 deletion(-) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py index 36b56c730bd..064c95ed35c 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py @@ -382,6 +382,13 @@ class DescriptorSpec: site_packages_linux=("cuquantum/lib",), dependencies=("cutensor", "cublas", "cusolver"), ), + DescriptorSpec( + name="custatevec", + packaged_with="other", + linux_sonames=("libcustatevec.so.1",), + site_packages_linux=("cuquantum/lib",), + dependencies=("cublas", "cublasLt"), + ), DescriptorSpec( name="nccl", packaged_with="other", diff --git a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py index 36aaf65cb93..1368d0260cb 100644 --- a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py @@ -192,6 +192,15 @@ class HeaderDescriptorSpec: conda_targets_layout=False, use_ctk_root_canary=False, ), + HeaderDescriptorSpec( + name="custatevec", + packaged_with="other", + header_basename="custatevec.h", + site_packages_dirs=("cuquantum/include",), + available_on_windows=False, + conda_targets_layout=False, + use_ctk_root_canary=False, + ), HeaderDescriptorSpec( name="cutlass", packaged_with="other", diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index a6621ab5bb9..153a4ec3089 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -23,6 +23,7 @@ cu12 = [ "cuda-toolkit[cufile]==12.*; sys_platform != 'win32'", "cutensor-cu12", "cutensornet-cu12; sys_platform != 'win32'", + "custatevec-cu12; sys_platform != 'win32'", "nvidia-cublasmp-cu12; sys_platform != 'win32'", "nvidia-cudss-cu12", "nvidia-cufftmp-cu12; sys_platform != 'win32'", @@ -37,6 +38,7 @@ cu13 = [ "cuda-toolkit[cufile]==13.*; sys_platform != 'win32'", "cutensor-cu13", "cutensornet-cu13; sys_platform != 'win32'", + "custatevec-cu13; sys_platform != 'win32'", "nvidia-cublasmp-cu13; sys_platform != 'win32'", "nvidia-cudla; platform_system == 'Linux' and platform_machine == 'aarch64'", "nvidia-cudss-cu13", diff --git a/cuda_pathfinder/tests/test_find_nvidia_headers.py b/cuda_pathfinder/tests/test_find_nvidia_headers.py index dc30feae2b9..819929ceab7 100644 --- a/cuda_pathfinder/tests/test_find_nvidia_headers.py +++ b/cuda_pathfinder/tests/test_find_nvidia_headers.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 # Currently these installations are only manually tested: @@ -47,6 +47,7 @@ "cute": r"^nvidia-cutlass$", "cutensor": r"^cutensor-.*$", "cutensornet": r"^cutensornet-.*$", + "custatevec": r"^custatevec-.*$", "cutlass": r"^nvidia-cutlass$", "mathdx": r"^nvidia-libmathdx-.*$", "nvshmem": r"^nvidia-nvshmem-.*$", diff --git a/toolshed/conda_create_for_pathfinder_testing.sh b/toolshed/conda_create_for_pathfinder_testing.sh index d6809c1cee9..28f40a633a1 100755 --- a/toolshed/conda_create_for_pathfinder_testing.sh +++ b/toolshed/conda_create_for_pathfinder_testing.sh @@ -28,6 +28,7 @@ cpkgs=( "cusparselt-dev" "cutensor" "cutensornet" + "custatevec" "cutlass" "libcublasmp-dev" "libcudss-dev" From 7b2fa01aaf553d44ac2774495389485220970c22 Mon Sep 17 00:00:00 2001 From: yangg Date: Wed, 15 Jul 2026 17:34:11 -0700 Subject: [PATCH 3/6] Pathfinder: add support on cudensitymat for library loading and header descriptors --- .../_dynamic_libs/descriptor_catalog.py | 16 ++++++++++++++++ .../_headers/header_descriptor_catalog.py | 9 +++++++++ cuda_pathfinder/pyproject.toml | 2 ++ .../tests/test_find_nvidia_headers.py | 1 + toolshed/conda_create_for_pathfinder_testing.sh | 1 + 5 files changed, 29 insertions(+) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py index 064c95ed35c..fe70c08c7a4 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py @@ -375,6 +375,22 @@ class DescriptorSpec: site_packages_windows=("cutensor/bin",), dependencies=("cutensor", "cublasLt"), ), + DescriptorSpec( + name="cudensitymat", + packaged_with="other", + linux_sonames=("libcudensitymat.so.0",), + site_packages_linux=("cuquantum/lib",), + dependencies=( + "cutensornet", + "cutensor", + "cusolver", + "cublasLt", + "cublas", + "curand", + "cusparse", + "nvJitLink", + ), + ), DescriptorSpec( name="cutensornet", packaged_with="other", diff --git a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py index 1368d0260cb..13a7584e08a 100644 --- a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py @@ -183,6 +183,15 @@ class HeaderDescriptorSpec: conda_targets_layout=False, use_ctk_root_canary=False, ), + HeaderDescriptorSpec( + name="cudensitymat", + packaged_with="other", + header_basename="cudensitymat.h", + site_packages_dirs=("cuquantum/include",), + available_on_windows=False, + conda_targets_layout=False, + use_ctk_root_canary=False, + ), HeaderDescriptorSpec( name="cutensornet", packaged_with="other", diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index 153a4ec3089..f32bc38e13a 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -21,6 +21,7 @@ test = [ cu12 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*", "cuda-toolkit[cufile]==12.*; sys_platform != 'win32'", + "cudensitymat-cu12; sys_platform != 'win32'", "cutensor-cu12", "cutensornet-cu12; sys_platform != 'win32'", "custatevec-cu12; sys_platform != 'win32'", @@ -36,6 +37,7 @@ cu12 = [ cu13 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*", "cuda-toolkit[cufile]==13.*; sys_platform != 'win32'", + "cudensitymat-cu13; sys_platform != 'win32'", "cutensor-cu13", "cutensornet-cu13; sys_platform != 'win32'", "custatevec-cu13; sys_platform != 'win32'", diff --git a/cuda_pathfinder/tests/test_find_nvidia_headers.py b/cuda_pathfinder/tests/test_find_nvidia_headers.py index 819929ceab7..fbdf0daf484 100644 --- a/cuda_pathfinder/tests/test_find_nvidia_headers.py +++ b/cuda_pathfinder/tests/test_find_nvidia_headers.py @@ -42,6 +42,7 @@ assert STRICTNESS in ("see_what_works", "all_must_work") NON_CTK_IMPORTLIB_METADATA_DISTRIBUTIONS_NAMES = { + "cudensitymat": r"^cudensitymat-.*$", "cusolverMp": r"^nvidia-cusolvermp-.*$", "cusparseLt": r"^nvidia-cusparselt-.*$", "cute": r"^nvidia-cutlass$", diff --git a/toolshed/conda_create_for_pathfinder_testing.sh b/toolshed/conda_create_for_pathfinder_testing.sh index 28f40a633a1..76c2c9154ad 100755 --- a/toolshed/conda_create_for_pathfinder_testing.sh +++ b/toolshed/conda_create_for_pathfinder_testing.sh @@ -25,6 +25,7 @@ set -u # Keep this list aligned with the Linux-installable subset of # cuda_pathfinder/pyproject.toml. cpkgs=( + "cudensitymat" "cusparselt-dev" "cutensor" "cutensornet" From d4db8d1f240ce7a2e64a63b903899fd84782ab3d Mon Sep 17 00:00:00 2001 From: yangg Date: Wed, 15 Jul 2026 17:39:35 -0700 Subject: [PATCH 4/6] Pathfinder: add support on custabilizer for library loading and header descriptors --- .../cuda/pathfinder/_dynamic_libs/descriptor_catalog.py | 6 ++++++ .../pathfinder/_headers/header_descriptor_catalog.py | 9 +++++++++ cuda_pathfinder/pyproject.toml | 2 ++ cuda_pathfinder/tests/test_find_nvidia_headers.py | 1 + toolshed/conda_create_for_pathfinder_testing.sh | 1 + 5 files changed, 19 insertions(+) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py index fe70c08c7a4..aeb22838544 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py @@ -398,6 +398,12 @@ class DescriptorSpec: site_packages_linux=("cuquantum/lib",), dependencies=("cutensor", "cublas", "cusolver"), ), + DescriptorSpec( + name="custabilizer", + packaged_with="other", + linux_sonames=("libcustabilizer.so.0",), + site_packages_linux=("cuquantum/lib",), + ), DescriptorSpec( name="custatevec", packaged_with="other", diff --git a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py index 13a7584e08a..6a3c68b580b 100644 --- a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py @@ -201,6 +201,15 @@ class HeaderDescriptorSpec: conda_targets_layout=False, use_ctk_root_canary=False, ), + HeaderDescriptorSpec( + name="custabilizer", + packaged_with="other", + header_basename="custabilizer.h", + site_packages_dirs=("cuquantum/include",), + available_on_windows=False, + conda_targets_layout=False, + use_ctk_root_canary=False, + ), HeaderDescriptorSpec( name="custatevec", packaged_with="other", diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index f32bc38e13a..fb05616dbc8 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -24,6 +24,7 @@ cu12 = [ "cudensitymat-cu12; sys_platform != 'win32'", "cutensor-cu12", "cutensornet-cu12; sys_platform != 'win32'", + "custabilizer-cu12; sys_platform != 'win32'", "custatevec-cu12; sys_platform != 'win32'", "nvidia-cublasmp-cu12; sys_platform != 'win32'", "nvidia-cudss-cu12", @@ -40,6 +41,7 @@ cu13 = [ "cudensitymat-cu13; sys_platform != 'win32'", "cutensor-cu13", "cutensornet-cu13; sys_platform != 'win32'", + "custabilizer-cu13; sys_platform != 'win32'", "custatevec-cu13; sys_platform != 'win32'", "nvidia-cublasmp-cu13; sys_platform != 'win32'", "nvidia-cudla; platform_system == 'Linux' and platform_machine == 'aarch64'", diff --git a/cuda_pathfinder/tests/test_find_nvidia_headers.py b/cuda_pathfinder/tests/test_find_nvidia_headers.py index fbdf0daf484..fcc5415316e 100644 --- a/cuda_pathfinder/tests/test_find_nvidia_headers.py +++ b/cuda_pathfinder/tests/test_find_nvidia_headers.py @@ -48,6 +48,7 @@ "cute": r"^nvidia-cutlass$", "cutensor": r"^cutensor-.*$", "cutensornet": r"^cutensornet-.*$", + "custabilizer": r"^custabilizer-.*$", "custatevec": r"^custatevec-.*$", "cutlass": r"^nvidia-cutlass$", "mathdx": r"^nvidia-libmathdx-.*$", diff --git a/toolshed/conda_create_for_pathfinder_testing.sh b/toolshed/conda_create_for_pathfinder_testing.sh index 76c2c9154ad..c2d2f45eda2 100755 --- a/toolshed/conda_create_for_pathfinder_testing.sh +++ b/toolshed/conda_create_for_pathfinder_testing.sh @@ -29,6 +29,7 @@ cpkgs=( "cusparselt-dev" "cutensor" "cutensornet" + "custabilizer" "custatevec" "cutlass" "libcublasmp-dev" From c2f28f524bd11da2f4be912f97a365b70cfc539a Mon Sep 17 00:00:00 2001 From: yangg Date: Wed, 15 Jul 2026 18:03:10 -0700 Subject: [PATCH 5/6] Pathfinder: add support on cupauliprop for library loading and header descriptors --- .../cuda/pathfinder/_dynamic_libs/descriptor_catalog.py | 6 ++++++ .../pathfinder/_headers/header_descriptor_catalog.py | 9 +++++++++ cuda_pathfinder/pyproject.toml | 2 ++ cuda_pathfinder/tests/test_find_nvidia_headers.py | 1 + toolshed/conda_create_for_pathfinder_testing.sh | 1 + 5 files changed, 19 insertions(+) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py index aeb22838544..bb75c5387b1 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py @@ -391,6 +391,12 @@ class DescriptorSpec: "nvJitLink", ), ), + DescriptorSpec( + name="cupauliprop", + packaged_with="other", + linux_sonames=("libcupauliprop.so.0",), + site_packages_linux=("cuquantum/lib",), + ), DescriptorSpec( name="cutensornet", packaged_with="other", diff --git a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py index 6a3c68b580b..b364e224e7e 100644 --- a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py @@ -192,6 +192,15 @@ class HeaderDescriptorSpec: conda_targets_layout=False, use_ctk_root_canary=False, ), + HeaderDescriptorSpec( + name="cupauliprop", + packaged_with="other", + header_basename="cupauliprop.h", + site_packages_dirs=("cuquantum/include",), + available_on_windows=False, + conda_targets_layout=False, + use_ctk_root_canary=False, + ), HeaderDescriptorSpec( name="cutensornet", packaged_with="other", diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index fb05616dbc8..ad382450e23 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -22,6 +22,7 @@ cu12 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*", "cuda-toolkit[cufile]==12.*; sys_platform != 'win32'", "cudensitymat-cu12; sys_platform != 'win32'", + "cupauliprop-cu12; sys_platform != 'win32'", "cutensor-cu12", "cutensornet-cu12; sys_platform != 'win32'", "custabilizer-cu12; sys_platform != 'win32'", @@ -39,6 +40,7 @@ cu13 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*", "cuda-toolkit[cufile]==13.*; sys_platform != 'win32'", "cudensitymat-cu13; sys_platform != 'win32'", + "cupauliprop-cu13; sys_platform != 'win32'", "cutensor-cu13", "cutensornet-cu13; sys_platform != 'win32'", "custabilizer-cu13; sys_platform != 'win32'", diff --git a/cuda_pathfinder/tests/test_find_nvidia_headers.py b/cuda_pathfinder/tests/test_find_nvidia_headers.py index fcc5415316e..90fe3cf9815 100644 --- a/cuda_pathfinder/tests/test_find_nvidia_headers.py +++ b/cuda_pathfinder/tests/test_find_nvidia_headers.py @@ -43,6 +43,7 @@ NON_CTK_IMPORTLIB_METADATA_DISTRIBUTIONS_NAMES = { "cudensitymat": r"^cudensitymat-.*$", + "cupauliprop": r"^cupauliprop-.*$", "cusolverMp": r"^nvidia-cusolvermp-.*$", "cusparseLt": r"^nvidia-cusparselt-.*$", "cute": r"^nvidia-cutlass$", diff --git a/toolshed/conda_create_for_pathfinder_testing.sh b/toolshed/conda_create_for_pathfinder_testing.sh index c2d2f45eda2..42f62c4e69c 100755 --- a/toolshed/conda_create_for_pathfinder_testing.sh +++ b/toolshed/conda_create_for_pathfinder_testing.sh @@ -26,6 +26,7 @@ set -u # cuda_pathfinder/pyproject.toml. cpkgs=( "cudensitymat" + "cupauliprop" "cusparselt-dev" "cutensor" "cutensornet" From 4fca19166be371510f9a9b2aaf668de179e046e7 Mon Sep 17 00:00:00 2001 From: yangg Date: Wed, 15 Jul 2026 18:12:36 -0700 Subject: [PATCH 6/6] switch to cuquantum metapackage to simplify dependency listing --- cuda_pathfinder/pyproject.toml | 12 ++---------- toolshed/conda_create_for_pathfinder_testing.sh | 6 +----- 2 files changed, 3 insertions(+), 15 deletions(-) diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index ad382450e23..227b8fa4eb7 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -21,12 +21,8 @@ test = [ cu12 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*", "cuda-toolkit[cufile]==12.*; sys_platform != 'win32'", - "cudensitymat-cu12; sys_platform != 'win32'", - "cupauliprop-cu12; sys_platform != 'win32'", + "cuquantum-cu12; sys_platform != 'win32'", "cutensor-cu12", - "cutensornet-cu12; sys_platform != 'win32'", - "custabilizer-cu12; sys_platform != 'win32'", - "custatevec-cu12; sys_platform != 'win32'", "nvidia-cublasmp-cu12; sys_platform != 'win32'", "nvidia-cudss-cu12", "nvidia-cufftmp-cu12; sys_platform != 'win32'", @@ -39,12 +35,8 @@ cu12 = [ cu13 = [ "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*", "cuda-toolkit[cufile]==13.*; sys_platform != 'win32'", - "cudensitymat-cu13; sys_platform != 'win32'", - "cupauliprop-cu13; sys_platform != 'win32'", + "cuquantum-cu13; sys_platform != 'win32'", "cutensor-cu13", - "cutensornet-cu13; sys_platform != 'win32'", - "custabilizer-cu13; sys_platform != 'win32'", - "custatevec-cu13; sys_platform != 'win32'", "nvidia-cublasmp-cu13; sys_platform != 'win32'", "nvidia-cudla; platform_system == 'Linux' and platform_machine == 'aarch64'", "nvidia-cudss-cu13", diff --git a/toolshed/conda_create_for_pathfinder_testing.sh b/toolshed/conda_create_for_pathfinder_testing.sh index 42f62c4e69c..4e38c5dbe88 100755 --- a/toolshed/conda_create_for_pathfinder_testing.sh +++ b/toolshed/conda_create_for_pathfinder_testing.sh @@ -25,13 +25,9 @@ set -u # Keep this list aligned with the Linux-installable subset of # cuda_pathfinder/pyproject.toml. cpkgs=( - "cudensitymat" - "cupauliprop" + "cuquantum" "cusparselt-dev" "cutensor" - "cutensornet" - "custabilizer" - "custatevec" "cutlass" "libcublasmp-dev" "libcudss-dev"