Skip to content

Commit 2715310

Browse files
committed
Also avoid use of cuda-toolkit[...] meta-package in cu12-linux and cu13-linux, to resolve a regression observed under linux-64
1 parent fa384e3 commit 2715310

File tree

2 files changed

+4
-111
lines changed

2 files changed

+4
-111
lines changed

cuda_pathfinder/pixi.lock

Lines changed: 1 addition & 109 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

cuda_pathfinder/pyproject.toml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ dependencies = []
1616
# therefore we are forced to break out subgroups by markers.
1717
test = ["pytest>=6.2.4"]
1818
cu12-common = [
19+
# Cannot use "cuda-toolkit[...]==12.* here, it confuses pixi (observed with pixi 0.56.0)
1920
"nvidia-cublas-cu12",
2021
"nvidia-cuda-cccl-cu12",
2122
"nvidia-cuda-nvcc-cu12",
@@ -33,7 +34,7 @@ cu12-common = [
3334
"nvidia-libmathdx-cu12",
3435
]
3536
cu12-linux = [
36-
"cuda-toolkit[cufile]==12.*; sys_platform != 'win32'",
37+
"nvidia-cufile-cu12; sys_platform != 'win32'",
3738
"nvidia-cufftmp-cu12; sys_platform != 'win32'",
3839
"nvidia-nccl-cu12; sys_platform != 'win32'",
3940
"nvidia-nvshmem-cu12; sys_platform != 'win32'",
@@ -59,7 +60,7 @@ cu13-common = [
5960
"nvidia-cudss-cu13",
6061
]
6162
cu13-linux = [
62-
"cuda-toolkit[cufile]==13.*; sys_platform != 'win32'",
63+
"nvidia-cufile; sys_platform != 'win32'",
6364
"nvidia-cublasmp-cu13; sys_platform != 'win32'",
6465
"nvidia-nccl-cu13; sys_platform != 'win32'",
6566
"nvidia-nvshmem-cu13; sys_platform != 'win32'",

0 commit comments

Comments
 (0)