Skip to content

Commit b509ff5

Browse files
DEKHTIARJonathanDEKHTIARJonathan
committed
NVIDIA config added
1 parent a3d80d9 commit b509ff5

File tree

4 files changed

+147
-56
lines changed

4 files changed

+147
-56
lines changed

‎configs/torch-2.9/torch-2.9.0-variants.json‎

Lines changed: 14 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -11,19 +11,19 @@
1111
"enable-if": "platform_system == 'Linux' or platform_system == 'Windows'",
1212
"plugin-api": "nvidia_variant_provider.plugin:NvidiaVariantPlugin",
1313
"requires": [
14-
"nvidia-variant-provider>=0.0.1,<1.0.0"
14+
"nvidia-variant-provider>=0.0.2,<1.0.0"
1515
]
1616
},
1717
"intel": {
18-
"enable-if": "platform_system == 'Linux'",
18+
"enable-if": "platform_system == 'Linux' or platform_system == 'Windows'",
1919
"plugin-api": "intel_variant_provider.plugin:IntelVariantPlugin",
2020
"requires": [
2121
"intel-variant-provider>=0.0.2,<1.0.0"
2222
]
2323
}
2424
},
2525
"variants": {
26-
"00000000": {},
26+
"null": {},
2727
"cu126": {
2828
"nvidia": {
2929
"cuda_version_lower_bound": [
@@ -43,33 +43,32 @@
4343
"cu128": {
4444
"nvidia": {
4545
"cuda_version_lower_bound": [
46-
"12.8"
46+
"12.0"
4747
],
4848
"sm_arch": [
49-
"100_real",
50-
"120_real",
5149
"70_real",
5250
"75_real",
5351
"80_real",
5452
"86_real",
55-
"90_real"
53+
"90_real",
54+
"100_real",
55+
"120_real"
5656
]
5757
}
5858
},
59-
"cu129": {
59+
"cu13": {
6060
"nvidia": {
6161
"cuda_version_lower_bound": [
62-
"12.9"
62+
"13.0"
6363
],
6464
"sm_arch": [
65-
"100_real",
66-
"120_real",
67-
"120_virtual",
68-
"70_real",
6965
"75_real",
7066
"80_real",
7167
"86_real",
72-
"90_real"
68+
"90_real",
69+
"100_real",
70+
"120_real",
71+
"120_virtual"
7372
]
7473
}
7574
},
@@ -86,4 +85,4 @@
8685
}
8786
}
8887
}
89-
}
88+
}

‎configs/torch-2.9/torch_pyproject.toml‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,11 @@
33
namespace = ["nvidia", "intel"]
44

55
[variant.providers.nvidia]
6-
requires = ["nvidia-variant-provider>=0.0.1,<1.0.0"]
6+
requires = ["nvidia-variant-provider>=0.0.2,<1.0.0"]
77
plugin-api = "nvidia_variant_provider.plugin:NvidiaVariantPlugin"
88
enable-if = "platform_system == 'Linux' or platform_system == 'Windows'"
99

1010
[variant.providers.intel]
11-
requires = ["intel-variant-provider"]
12-
enable-if = "platform_system == 'Linux'"
11+
requires = ["intel-variant-provider>=0.0.2,<1.0.0"]
12+
enable-if = "platform_system == 'Linux' or platform_system == 'Windows'"
1313
plugin-api = "intel_variant_provider.plugin:IntelVariantPlugin"

‎configs/torch-2.9/torch_variant_config.toml‎

Lines changed: 116 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -27,28 +27,78 @@ deps_remove_list = [
2727
'umf',
2828
]
2929
deps_add_list = [
30+
# Common to CUDA 12 builds
31+
"triton==3.4.0; platform_system == 'Linux' and 'nvidia' in variant_namespaces",
32+
"nvidia-cudnn==9.10.2.21; platform_system == 'Linux' and 'nvidia :: cuda_version_lower_bound :: 12.0' in variant_properties",
33+
"nvidia-cusparselt==0.7.1; platform_system == 'Linux' and 'nvidia :: cuda_version_lower_bound :: 12.0' in variant_properties",
34+
"nvidia-nccl==2.27.5; platform_system == 'Linux' and 'nvidia :: cuda_version_lower_bound :: 12.0' in variant_properties",
35+
"nvidia-nvshmem==3.3.20; platform_system == 'Linux' and 'nvidia :: cuda_version_lower_bound :: 12.0' in variant_properties",
36+
37+
# CUDA 12.6
38+
"nvidia-cublas==12.6.4.1; platform_system == 'Linux' and variant_label == 'cu126'",
39+
"nvidia-cuda-cupti==12.6.80; platform_system == 'Linux' and variant_label == 'cu126'",
40+
"nvidia-cuda-nvrtc==12.6.77; platform_system == 'Linux' and variant_label == 'cu126'",
41+
"nvidia-cuda-runtime==12.6.77; platform_system == 'Linux' and variant_label == 'cu126'",
42+
"nvidia-cufft==11.3.0.4; platform_system == 'Linux' and variant_label == 'cu126'",
43+
"nvidia-cufile==1.11.1.6; platform_system == 'Linux' and variant_label == 'cu126'",
44+
"nvidia-curand==10.3.7.77; platform_system == 'Linux' and variant_label == 'cu126'",
45+
"nvidia-cusolver==11.7.1.2; platform_system == 'Linux' and variant_label == 'cu126'",
46+
"nvidia-cusparse==12.5.4.2; platform_system == 'Linux' and variant_label == 'cu126'",
47+
"nvidia-nvjitlink==12.6.85; platform_system == 'Linux' and variant_label == 'cu126'",
48+
"nvidia-nvtx==12.6.77; platform_system == 'Linux' and variant_label == 'cu126'",
49+
50+
# CUDA 12.8
51+
"nvidia-cublas==12.8.4.1; platform_system == 'Linux' and variant_label == 'cu128'",
52+
"nvidia-cuda-cupti==12.8.90; platform_system == 'Linux' and variant_label == 'cu128'",
53+
"nvidia-cuda-nvrtc==12.8.93; platform_system == 'Linux' and variant_label == 'cu128'",
54+
"nvidia-cuda-runtime==12.8.90; platform_system == 'Linux' and variant_label == 'cu128'",
55+
"nvidia-cufft==11.3.3.83; platform_system == 'Linux' and variant_label == 'cu128'",
56+
"nvidia-cufile==1.13.1.3; platform_system == 'Linux' and variant_label == 'cu128'",
57+
"nvidia-curand==10.3.9.90; platform_system == 'Linux' and variant_label == 'cu128'",
58+
"nvidia-cusolver==11.7.3.90; platform_system == 'Linux' and variant_label == 'cu128'",
59+
"nvidia-cusparse==12.5.8.93; platform_system == 'Linux' and variant_label == 'cu128'",
60+
"nvidia-nvjitlink==12.8.93; platform_system == 'Linux' and variant_label == 'cu128'",
61+
"nvidia-nvtx==12.8.90; platform_system == 'Linux' and variant_label == 'cu128'",
62+
63+
# CUDA 13
64+
"nvidia-cublas==13.0.0.19; platform_system == 'Linux' and variant_label == 'cu13'",
65+
"nvidia-cuda-cupti==13.0.48; platform_system == 'Linux' and variant_label == 'cu13'",
66+
"nvidia-cuda-nvrtc==13.0.48; platform_system == 'Linux' and variant_label == 'cu13'",
67+
"nvidia-cuda-runtime==13.0.48; platform_system == 'Linux' and variant_label == 'cu13'",
68+
"nvidia-cudnn==9.13.0.50; platform_system == 'Linux' and variant_label == 'cu13'",
69+
"nvidia-cufft==12.0.0.15; platform_system == 'Linux' and variant_label == 'cu13'",
70+
"nvidia-cufile==1.15.0.42; platform_system == 'Linux' and variant_label == 'cu13'",
71+
"nvidia-curand==10.4.0.35; platform_system == 'Linux' and variant_label == 'cu13'",
72+
"nvidia-cusolver==12.0.3.29; platform_system == 'Linux' and variant_label == 'cu13'",
73+
"nvidia-cusparse==12.6.2.49; platform_system == 'Linux' and variant_label == 'cu13'",
74+
"nvidia-cusparselt==0.8.0; platform_system == 'Linux' and variant_label == 'cu13'",
75+
"nvidia-nccl==2.27.7; platform_system == 'Linux' and variant_label == 'cu13'",
76+
"nvidia-nvjitlink==13.0.39; platform_system == 'Linux' and variant_label == 'cu13'",
77+
"nvidia-nvshmem==3.3.24; platform_system == 'Linux' and variant_label == 'cu13'",
78+
"nvidia-nvtx==13.0.39; platform_system == 'Linux' and variant_label == 'cu13'",
79+
3080
# XPU
31-
'dpcpp-cpp-rt==2025.2.1; platform_system == "Linux" and "intel" in variant_namespaces',
32-
'impi-rt==2021.16.1; platform_system == "Linux" and platform_machine == "x86_64" and "intel" in variant_namespaces',
33-
'intel-cmplr-lib-rt==2025.2.1; platform_system == "Linux" and "intel" in variant_namespaces',
34-
'intel-cmplr-lib-ur==2025.2.1; platform_system == "Linux" and "intel" in variant_namespaces',
35-
'intel-cmplr-lic-rt==2025.2.1; platform_system == "Linux" and "intel" in variant_namespaces',
36-
'intel-opencl-rt==2025.2.1; platform_system == "Linux" and "intel" in variant_namespaces',
37-
'intel-openmp==2025.2.1; platform_system == "Linux" and "intel" in variant_namespaces',
38-
'intel-pti==0.13.1; platform_system == "Linux" and "intel" in variant_namespaces',
39-
'intel-sycl-rt==2025.2.1; platform_system == "Linux" and "intel" in variant_namespaces',
40-
'mkl==2025.2.0; platform_system == "Linux" and "intel" in variant_namespaces',
41-
'oneccl-devel==2021.16.1; platform_system == "Linux" and platform_machine == "x86_64" and "intel" in variant_namespaces',
42-
'oneccl==2021.16.1; platform_system == "Linux" and platform_machine == "x86_64" and "intel" in variant_namespaces',
43-
'onemkl-sycl-blas==2025.2.0; platform_system == "Linux" and "intel" in variant_namespaces',
44-
'onemkl-sycl-dft==2025.2.0; platform_system == "Linux" and "intel" in variant_namespaces',
45-
'onemkl-sycl-lapack==2025.2.0; platform_system == "Linux" and "intel" in variant_namespaces',
46-
'onemkl-sycl-rng==2025.2.0; platform_system == "Linux" and "intel" in variant_namespaces',
47-
'onemkl-sycl-sparse==2025.2.0; platform_system == "Linux" and "intel" in variant_namespaces',
48-
'pytorch-triton-xpu==3.5.0; platform_system == "Linux" and "intel" in variant_namespaces',
49-
'tbb==2022.2.0; platform_system == "Linux" and "intel" in variant_namespaces',
50-
'tcmlib==1.4.0; platform_system == "Linux" and "intel" in variant_namespaces',
51-
'umf==0.11.0; platform_system == "Linux" and "intel" in variant_namespaces',
81+
"dpcpp-cpp-rt==2025.2.1; platform_system == 'Linux' and 'intel' in variant_namespaces",
82+
"impi-rt==2021.16.1; platform_system == 'Linux' and platform_machine == 'x86_64' and 'intel' in variant_namespaces",
83+
"intel-cmplr-lib-rt==2025.2.1; platform_system == 'Linux' and 'intel' in variant_namespaces",
84+
"intel-cmplr-lib-ur==2025.2.1; platform_system == 'Linux' and 'intel' in variant_namespaces",
85+
"intel-cmplr-lic-rt==2025.2.1; platform_system == 'Linux' and 'intel' in variant_namespaces",
86+
"intel-opencl-rt==2025.2.1; platform_system == 'Linux' and 'intel' in variant_namespaces",
87+
"intel-openmp==2025.2.1; platform_system == 'Linux' and 'intel' in variant_namespaces",
88+
"intel-pti==0.13.1; platform_system == 'Linux' and 'intel' in variant_namespaces",
89+
"intel-sycl-rt==2025.2.1; platform_system == 'Linux' and 'intel' in variant_namespaces",
90+
"mkl==2025.2.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
91+
"oneccl-devel==2021.16.1; platform_system == 'Linux' and platform_machine == 'x86_64' and 'intel' in variant_namespaces",
92+
"oneccl==2021.16.1; platform_system == 'Linux' and platform_machine == 'x86_64' and 'intel' in variant_namespaces",
93+
"onemkl-sycl-blas==2025.2.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
94+
"onemkl-sycl-dft==2025.2.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
95+
"onemkl-sycl-lapack==2025.2.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
96+
"onemkl-sycl-rng==2025.2.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
97+
"onemkl-sycl-sparse==2025.2.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
98+
"pytorch-triton-xpu==3.5.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
99+
"tbb==2022.2.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
100+
"tcmlib==1.4.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
101+
"umf==0.11.0; platform_system == 'Linux' and 'intel' in variant_namespaces",
52102
]
53103

54104
[metadata_configs.torchvision]
@@ -57,8 +107,51 @@ normalize_version = true
57107
deps_remove_list = ["torch"]
58108
deps_add_list = ["torch==2.9.0"]
59109

110+
111+
# PyT CUDA `sm_archs`: https://github.com/pytorch/pytorch/blob/v2.9.0-rc4/.ci/manywheel/build_cuda.sh
112+
113+
[variant_configs]
114+
[variant_configs.cu126]
115+
variant_label = "cu126" # [0-9a-z._]{1,16}
116+
properties = [
117+
{ namespace = "nvidia", feature = "cuda_version_lower_bound", value = "12.0" },
118+
{ namespace = "nvidia", feature = "sm_arch", value = "50_real" },
119+
{ namespace = "nvidia", feature = "sm_arch", value = "60_real" },
120+
{ namespace = "nvidia", feature = "sm_arch", value = "70_real" },
121+
{ namespace = "nvidia", feature = "sm_arch", value = "75_real" },
122+
{ namespace = "nvidia", feature = "sm_arch", value = "80_real" },
123+
{ namespace = "nvidia", feature = "sm_arch", value = "86_real" },
124+
{ namespace = "nvidia", feature = "sm_arch", value = "90_real" },
125+
]
126+
127+
[variant_configs.cu128]
128+
variant_label = "cu128" # [0-9a-z._]{1,16}
129+
properties = [
130+
{ namespace = "nvidia", feature = "cuda_version_lower_bound", value = "12.0" },
131+
{ namespace = "nvidia", feature = "sm_arch", value = "70_real" },
132+
{ namespace = "nvidia", feature = "sm_arch", value = "75_real" },
133+
{ namespace = "nvidia", feature = "sm_arch", value = "80_real" },
134+
{ namespace = "nvidia", feature = "sm_arch", value = "86_real" },
135+
{ namespace = "nvidia", feature = "sm_arch", value = "90_real" },
136+
{ namespace = "nvidia", feature = "sm_arch", value = "100_real" },
137+
{ namespace = "nvidia", feature = "sm_arch", value = "120_real" },
138+
]
139+
140+
[variant_configs.cu13]
141+
variant_label = "cu13" # [0-9a-z._]{1,16}
142+
properties = [
143+
{ namespace = "nvidia", feature = "cuda_version_lower_bound", value = "13.0" },
144+
{ namespace = "nvidia", feature = "sm_arch", value = "75_real" },
145+
{ namespace = "nvidia", feature = "sm_arch", value = "80_real" },
146+
{ namespace = "nvidia", feature = "sm_arch", value = "86_real" },
147+
{ namespace = "nvidia", feature = "sm_arch", value = "90_real" },
148+
{ namespace = "nvidia", feature = "sm_arch", value = "100_real" },
149+
{ namespace = "nvidia", feature = "sm_arch", value = "120_real" },
150+
{ namespace = "nvidia", feature = "sm_arch", value = "120_virtual" },
151+
]
152+
60153
[variant_configs.intel]
61-
variant_label = "xpu" # [a-z0-9_]{1,8}
154+
variant_label = "xpu" # [0-9a-z._]{1,16}
62155
properties = [
63156
{ namespace = "intel", feature = "device_ip", value = "20.4.4" }, # lnl-m
64157
{ namespace = "intel", feature = "device_ip", value = "20.1.0" }, # bmg
@@ -69,5 +162,5 @@ properties = [
69162
]
70163

71164
[variant_configs.cpu]
72-
# variant_label = None # FORBIDDEN - No Alias -> NULL VARIANT: `00000000`
165+
# variant_label = None # FORBIDDEN - No Alias -> NULL VARIANT: `null`
73166
properties = []

‎configs/torch-2.9/torchvision-0.24.0-variants.json‎

Lines changed: 14 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -11,19 +11,19 @@
1111
"enable-if": "platform_system == 'Linux' or platform_system == 'Windows'",
1212
"plugin-api": "nvidia_variant_provider.plugin:NvidiaVariantPlugin",
1313
"requires": [
14-
"nvidia-variant-provider>=0.0.1,<1.0.0"
14+
"nvidia-variant-provider>=0.0.2,<1.0.0"
1515
]
1616
},
1717
"intel": {
18-
"enable-if": "platform_system == 'Linux'",
18+
"enable-if": "platform_system == 'Linux' or platform_system == 'Windows'",
1919
"plugin-api": "intel_variant_provider.plugin:IntelVariantPlugin",
2020
"requires": [
2121
"intel-variant-provider>=0.0.2,<1.0.0"
2222
]
2323
}
2424
},
2525
"variants": {
26-
"00000000": {},
26+
"null": {},
2727
"cu126": {
2828
"nvidia": {
2929
"cuda_version_lower_bound": [
@@ -43,33 +43,32 @@
4343
"cu128": {
4444
"nvidia": {
4545
"cuda_version_lower_bound": [
46-
"12.8"
46+
"12.0"
4747
],
4848
"sm_arch": [
49-
"100_real",
50-
"120_real",
5149
"70_real",
5250
"75_real",
5351
"80_real",
5452
"86_real",
55-
"90_real"
53+
"90_real",
54+
"100_real",
55+
"120_real"
5656
]
5757
}
5858
},
59-
"cu129": {
59+
"cu13": {
6060
"nvidia": {
6161
"cuda_version_lower_bound": [
62-
"12.9"
62+
"13.0"
6363
],
6464
"sm_arch": [
65-
"100_real",
66-
"120_real",
67-
"120_virtual",
68-
"70_real",
6965
"75_real",
7066
"80_real",
7167
"86_real",
72-
"90_real"
68+
"90_real",
69+
"100_real",
70+
"120_real",
71+
"120_virtual"
7372
]
7473
}
7574
},
@@ -86,4 +85,4 @@
8685
}
8786
}
8887
}
89-
}
88+
}

0 commit comments

Comments
 (0)