Skip to content

Commit 597b67f

Browse files
committed
chore: factor out shared cu dependencies
1 parent 52b15cc commit 597b67f

File tree

1 file changed

+13
-12
lines changed

1 file changed

+13
-12
lines changed

pixi.toml

Lines changed: 13 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,7 @@ cuda-compiler = "*"
2020
make = "*"
2121
cxx-compiler = "*"
2222

23-
[feature.cu-12.system-requirements]
24-
cuda = "12"
25-
26-
[feature.cu-12.dependencies]
27-
cuda-version = "12.*"
23+
[feature.cu.dependencies]
2824
cuda-nvcc = "*"
2925
cuda-nvcc-impl = "*"
3026
cuda-cuobjdump = "*"
@@ -33,6 +29,12 @@ libnvjitlink = "*"
3329
cuda-cccl = "*"
3430
libcurand = "*"
3531

32+
[feature.cu-12.system-requirements]
33+
cuda = "12"
34+
35+
[feature.cu-12.dependencies]
36+
cuda-version = "12.*"
37+
3638
[feature.cu-12-0.dependencies]
3739
cuda-version = "12.0.*"
3840
# cuda-runtime fails to solve because of nvjitlink
@@ -57,14 +59,8 @@ cuda = "13"
5759

5860
[feature.cu-13.dependencies]
5961
cuda-version = "13.*"
60-
cuda-nvvm = "*"
61-
cuda-nvrtc = "*"
62-
libnvjitlink = "*"
63-
cuda-cccl = "*"
64-
libcurand = "*"
6562
cuda-runtime = "*"
66-
cuda-nvcc-impl = "*"
67-
cuda-cuobjdump = "*"
63+
cuda-nvvm = "*"
6864

6965
[feature.cu-13-0.dependencies]
7066

@@ -92,32 +88,37 @@ cu-12-0 = { features = [
9288
"cu-12-0",
9389
"test",
9490
"build",
91+
"cu",
9592
"cu-12",
9693
], solve-group = "cu-12-0" }
9794
cu-12-2 = { features = [
9895
"cu-12-2",
9996
"test",
10097
"build",
98+
"cu",
10199
"cu-12",
102100
], solve-group = "cu-12-2" }
103101
cu-12-8 = { features = [
104102
"cu-12-8",
105103
"test",
106104
"build",
105+
"cu",
107106
"cu-12",
108107
], solve-group = "cu-12-8" }
109108
cu-12-9 = { features = [
110109
"cu-12-9",
111110
"test",
112111
"build",
113112
"bench",
113+
"cu",
114114
"cu-12",
115115
], solve-group = "cu-12-9" }
116116
# CUDA 13
117117
cu-13-0 = { features = [
118118
"cu-13-0",
119119
"test",
120120
"build",
121+
"cu",
121122
"cu-13",
122123
], solve-group = "cu-13-0" }
123124

0 commit comments

Comments
 (0)