From 1d37c99ba2cfea3a9b1b4297ec0f95ff2cb80a16 Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Wed, 8 Apr 2026 23:25:45 -0700 Subject: [PATCH] Exclude broken cuda-toolkit wheels on Windows. Made-with: Cursor --- cuda_bindings/pyproject.toml | 3 ++- cuda_core/pyproject.toml | 30 ++++++++++++++++++++++++++---- cuda_pathfinder/pyproject.toml | 6 ++++-- 3 files changed, 32 insertions(+), 7 deletions(-) diff --git a/cuda_bindings/pyproject.toml b/cuda_bindings/pyproject.toml index f4866fc4f8..6242aeb1c6 100644 --- a/cuda_bindings/pyproject.toml +++ b/cuda_bindings/pyproject.toml @@ -35,7 +35,8 @@ dependencies = ["cuda-pathfinder >=1.4.2"] [project.optional-dependencies] all = [ - "cuda-toolkit[nvrtc,nvjitlink,nvvm,nvfatbin]==13.*", + "cuda-toolkit[nvrtc,nvjitlink,nvvm,nvfatbin]==13.*,!=13.0.3; sys_platform == 'win32'", + "cuda-toolkit[nvrtc,nvjitlink,nvvm,nvfatbin]==13.*; sys_platform != 'win32'", "cuda-toolkit[cufile]==13.*; sys_platform == 'linux'", ] diff --git a/cuda_core/pyproject.toml b/cuda_core/pyproject.toml index 80711f39ed..5c33302e0a 100644 --- a/cuda_core/pyproject.toml +++ b/cuda_core/pyproject.toml @@ -58,12 +58,34 @@ cu13 = ["cuda-bindings[all]==13.*"] [dependency-groups] test = ["cython>=3.2,<3.3", "setuptools", "pytest>=6.2.4", "pytest-benchmark", "pytest-randomly", "pytest-repeat", "pytest-rerunfailures", "cloudpickle", "psutil", "cffi"] ml-dtypes = ["ml-dtypes>=0.5.4,<0.6.0"] -test-cu12 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda12x; python_version < '3.14'", "cuda-toolkit[cudart]==12.*"] # runtime headers needed by CuPy -test-cu13 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda13x; python_version < '3.14'", "cuda-toolkit[cudart]==13.*"] # runtime headers needed by CuPy +test-cu12 = [ + {include-group = "ml-dtypes" }, + {include-group = "test" }, + "cupy-cuda12x; python_version < '3.14'", + "cuda-toolkit[cudart]==12.*,!=12.9.2; sys_platform == 'win32'", + "cuda-toolkit[cudart]==12.*; sys_platform != 'win32'", +] # runtime headers needed by CuPy +test-cu13 = [ + {include-group = "ml-dtypes" }, + {include-group = "test" }, + "cupy-cuda13x; python_version < '3.14'", + "cuda-toolkit[cudart]==13.*,!=13.0.3; sys_platform == 'win32'", + "cuda-toolkit[cudart]==13.*; sys_platform != 'win32'", +] # runtime headers needed by CuPy # free threaded build, cupy doesn't support free-threaded builds yet, so avoid installing it for now # TODO: cupy should support free threaded builds -test-cu12-ft = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cuda-toolkit[cudart]==12.*"] -test-cu13-ft = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cuda-toolkit[cudart]==13.*"] +test-cu12-ft = [ + {include-group = "ml-dtypes" }, + {include-group = "test" }, + "cuda-toolkit[cudart]==12.*,!=12.9.2; sys_platform == 'win32'", + "cuda-toolkit[cudart]==12.*; sys_platform != 'win32'", +] +test-cu13-ft = [ + {include-group = "ml-dtypes" }, + {include-group = "test" }, + "cuda-toolkit[cudart]==13.*,!=13.0.3; sys_platform == 'win32'", + "cuda-toolkit[cudart]==13.*; sys_platform != 'win32'", +] [tool.uv] conflicts = [ diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index 7d96e72023..fe6383aa79 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -19,7 +19,8 @@ test = [ ] # Internal organization of test dependencies. cu12 = [ - "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*", + "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*,!=12.9.2; sys_platform == 'win32'", + "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*; sys_platform != 'win32'", "cuda-toolkit[cufile]==12.*; sys_platform != 'win32'", "cutensor-cu12", "nvidia-cublasmp-cu12; sys_platform != 'win32'", @@ -32,7 +33,8 @@ cu12 = [ "nvidia-nvshmem-cu12; sys_platform != 'win32'", ] cu13 = [ - "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*", + "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*,!=13.0.3; sys_platform == 'win32'", + "cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*; sys_platform != 'win32'", "cuda-toolkit[cufile]==13.*; sys_platform != 'win32'", "cutensor-cu13", "nvidia-cublasmp-cu13; sys_platform != 'win32'",