Skip to content

Commit 1d37c99

Browse files
committed
Exclude broken cuda-toolkit wheels on Windows.
Made-with: Cursor
1 parent c93623b commit 1d37c99

File tree

3 files changed

+32
-7
lines changed

3 files changed

+32
-7
lines changed

cuda_bindings/pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,8 @@ dependencies = ["cuda-pathfinder >=1.4.2"]
3535

3636
[project.optional-dependencies]
3737
all = [
38-
"cuda-toolkit[nvrtc,nvjitlink,nvvm,nvfatbin]==13.*",
38+
"cuda-toolkit[nvrtc,nvjitlink,nvvm,nvfatbin]==13.*,!=13.0.3; sys_platform == 'win32'",
39+
"cuda-toolkit[nvrtc,nvjitlink,nvvm,nvfatbin]==13.*; sys_platform != 'win32'",
3940
"cuda-toolkit[cufile]==13.*; sys_platform == 'linux'",
4041
]
4142

cuda_core/pyproject.toml

Lines changed: 26 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -58,12 +58,34 @@ cu13 = ["cuda-bindings[all]==13.*"]
5858
[dependency-groups]
5959
test = ["cython>=3.2,<3.3", "setuptools", "pytest>=6.2.4", "pytest-benchmark", "pytest-randomly", "pytest-repeat", "pytest-rerunfailures", "cloudpickle", "psutil", "cffi"]
6060
ml-dtypes = ["ml-dtypes>=0.5.4,<0.6.0"]
61-
test-cu12 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda12x; python_version < '3.14'", "cuda-toolkit[cudart]==12.*"] # runtime headers needed by CuPy
62-
test-cu13 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda13x; python_version < '3.14'", "cuda-toolkit[cudart]==13.*"] # runtime headers needed by CuPy
61+
test-cu12 = [
62+
{include-group = "ml-dtypes" },
63+
{include-group = "test" },
64+
"cupy-cuda12x; python_version < '3.14'",
65+
"cuda-toolkit[cudart]==12.*,!=12.9.2; sys_platform == 'win32'",
66+
"cuda-toolkit[cudart]==12.*; sys_platform != 'win32'",
67+
] # runtime headers needed by CuPy
68+
test-cu13 = [
69+
{include-group = "ml-dtypes" },
70+
{include-group = "test" },
71+
"cupy-cuda13x; python_version < '3.14'",
72+
"cuda-toolkit[cudart]==13.*,!=13.0.3; sys_platform == 'win32'",
73+
"cuda-toolkit[cudart]==13.*; sys_platform != 'win32'",
74+
] # runtime headers needed by CuPy
6375
# free threaded build, cupy doesn't support free-threaded builds yet, so avoid installing it for now
6476
# TODO: cupy should support free threaded builds
65-
test-cu12-ft = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cuda-toolkit[cudart]==12.*"]
66-
test-cu13-ft = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cuda-toolkit[cudart]==13.*"]
77+
test-cu12-ft = [
78+
{include-group = "ml-dtypes" },
79+
{include-group = "test" },
80+
"cuda-toolkit[cudart]==12.*,!=12.9.2; sys_platform == 'win32'",
81+
"cuda-toolkit[cudart]==12.*; sys_platform != 'win32'",
82+
]
83+
test-cu13-ft = [
84+
{include-group = "ml-dtypes" },
85+
{include-group = "test" },
86+
"cuda-toolkit[cudart]==13.*,!=13.0.3; sys_platform == 'win32'",
87+
"cuda-toolkit[cudart]==13.*; sys_platform != 'win32'",
88+
]
6789

6890
[tool.uv]
6991
conflicts = [

cuda_pathfinder/pyproject.toml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@ test = [
1919
]
2020
# Internal organization of test dependencies.
2121
cu12 = [
22-
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*",
22+
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*,!=12.9.2; sys_platform == 'win32'",
23+
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler]==12.*; sys_platform != 'win32'",
2324
"cuda-toolkit[cufile]==12.*; sys_platform != 'win32'",
2425
"cutensor-cu12",
2526
"nvidia-cublasmp-cu12; sys_platform != 'win32'",
@@ -32,7 +33,8 @@ cu12 = [
3233
"nvidia-nvshmem-cu12; sys_platform != 'win32'",
3334
]
3435
cu13 = [
35-
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*",
36+
"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'",
37+
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,cupti,profiler,nvvm]==13.*; sys_platform != 'win32'",
3638
"cuda-toolkit[cufile]==13.*; sys_platform != 'win32'",
3739
"cutensor-cu13",
3840
"nvidia-cublasmp-cu13; sys_platform != 'win32'",

0 commit comments

Comments
 (0)