Skip to content

Commit c164ecc

Browse files
dvrogozhdvrogozh
committed
configs: torch-3.9 xpu configs
Signed-off-by: Dmitry Rogozhkin <[email protected]>
1 parent af5d878 commit c164ecc

File tree

4 files changed

+264
-0
lines changed

4 files changed

+264
-0
lines changed

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

Lines changed: 89 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,89 @@
1+
{
2+
"$schema": "https://variants-schema.wheelnext.dev/v0.0.2.json",
3+
"default-priorities": {
4+
"namespace": [
5+
"nvidia",
6+
"intel"
7+
]
8+
},
9+
"providers": {
10+
"nvidia": {
11+
"enable-if": "platform_system == 'Linux' or platform_system == 'Windows'",
12+
"plugin-api": "nvidia_variant_provider.plugin:NvidiaVariantPlugin",
13+
"requires": [
14+
"nvidia-variant-provider>=0.0.1,<1.0.0"
15+
]
16+
},
17+
"intel": {
18+
"enable-if": "platform_system == 'Linux'",
19+
"plugin-api": "intel_variant_provider.plugin:IntelVariantPlugin",
20+
"requires": [
21+
"intel-variant-provider"
22+
]
23+
}
24+
},
25+
"variants": {
26+
"00000000": {},
27+
"cu126": {
28+
"nvidia": {
29+
"cuda_version_lower_bound": [
30+
"12.0"
31+
],
32+
"sm_arch": [
33+
"50_real",
34+
"60_real",
35+
"70_real",
36+
"75_real",
37+
"80_real",
38+
"86_real",
39+
"90_real"
40+
]
41+
}
42+
},
43+
"cu128": {
44+
"nvidia": {
45+
"cuda_version_lower_bound": [
46+
"12.8"
47+
],
48+
"sm_arch": [
49+
"100_real",
50+
"120_real",
51+
"70_real",
52+
"75_real",
53+
"80_real",
54+
"86_real",
55+
"90_real"
56+
]
57+
}
58+
},
59+
"cu129": {
60+
"nvidia": {
61+
"cuda_version_lower_bound": [
62+
"12.9"
63+
],
64+
"sm_arch": [
65+
"100_real",
66+
"120_real",
67+
"120_virtual",
68+
"70_real",
69+
"75_real",
70+
"80_real",
71+
"86_real",
72+
"90_real"
73+
]
74+
}
75+
},
76+
"xpu": {
77+
"intel": {
78+
"device_ip": [
79+
"12.55.8",
80+
"12.60.7",
81+
"12.71.4",
82+
"12.74.4",
83+
"20.1.0",
84+
"20.4.4"
85+
]
86+
}
87+
}
88+
}
89+
}

‎configs/torch-2.9/torch_pyproject.toml‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# https://github.com/pytorch/pytorch/blob/b46ca66793573f00f5bb7f8664ae72b4a5f38a2a/pyproject.toml
2+
[variant.default-priorities]
3+
namespace = ["nvidia", "intel"]
4+
5+
[variant.providers.nvidia]
6+
requires = ["nvidia-variant-provider>=0.0.1,<1.0.0"]
7+
plugin-api = "nvidia_variant_provider.plugin:NvidiaVariantPlugin"
8+
enable-if = "platform_system == 'Linux' or platform_system == 'Windows'"
9+
10+
[variant.providers.intel]
11+
requires = ["intel-variant-provider"]
12+
enable-if = "platform_system == 'Linux'"
13+
plugin-api = "intel_variant_provider.plugin:IntelVariantPlugin"

‎configs/torch-2.9/torch_variant_config.toml‎

Lines changed: 73 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,73 @@
1+
[metadata_configs]
2+
[metadata_configs.torch]
3+
normalize_package_name = false
4+
normalize_version = true
5+
deps_remove_list = [
6+
# XPU
7+
'dpcpp-cpp-rt',
8+
'impi-rt',
9+
'intel-cmplr-lib-rt',
10+
'intel-cmplr-lib-ur',
11+
'intel-cmplr-lic-rt',
12+
'intel-opencl-rt',
13+
'intel-openmp',
14+
'intel-pti',
15+
'intel-sycl-rt',
16+
'mkl',
17+
'oneccl-devel',
18+
'oneccl',
19+
'onemkl-sycl-blas',
20+
'onemkl-sycl-dft',
21+
'onemkl-sycl-lapack',
22+
'onemkl-sycl-rng',
23+
'onemkl-sycl-sparse',
24+
'pytorch-triton-xpu',
25+
'tbb',
26+
'tcmlib',
27+
'umf',
28+
]
29+
deps_add_list = [
30+
# 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',
52+
]
53+
54+
[metadata_configs.torchvision]
55+
normalize_package_name = false
56+
normalize_version = true
57+
deps_remove_list = ["torch"]
58+
deps_add_list = ["torch==2.9.0"]
59+
60+
[variant_configs.intel]
61+
variant_label = "xpu" # [a-z0-9_]{1,8}
62+
properties = [
63+
{ namespace = "intel", feature = "device_ip", value = "20.4.4" }, # lnl-m
64+
{ namespace = "intel", feature = "device_ip", value = "20.1.0" }, # bmg
65+
{ namespace = "intel", feature = "device_ip", value = "12.74.4" }, # arl-h
66+
{ namespace = "intel", feature = "device_ip", value = "12.71.4" }, # mtl-h
67+
{ namespace = "intel", feature = "device_ip", value = "12.60.7" }, # pvc
68+
{ namespace = "intel", feature = "device_ip", value = "12.55.8" }, # dg2
69+
]
70+
71+
[variant_configs.cpu]
72+
# variant_label = None # FORBIDDEN - No Alias -> NULL VARIANT: `00000000`
73+
properties = []

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

Lines changed: 89 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,89 @@
1+
{
2+
"$schema": "https://variants-schema.wheelnext.dev/v0.0.2.json",
3+
"default-priorities": {
4+
"namespace": [
5+
"nvidia",
6+
"intel"
7+
]
8+
},
9+
"providers": {
10+
"nvidia": {
11+
"enable-if": "platform_system == 'Linux' or platform_system == 'Windows'",
12+
"plugin-api": "nvidia_variant_provider.plugin:NvidiaVariantPlugin",
13+
"requires": [
14+
"nvidia-variant-provider>=0.0.1,<1.0.0"
15+
]
16+
},
17+
"intel": {
18+
"enable-if": "platform_system == 'Linux'",
19+
"plugin-api": "intel_variant_provider.plugin:IntelVariantPlugin",
20+
"requires": [
21+
"intel-variant-provider"
22+
]
23+
}
24+
},
25+
"variants": {
26+
"00000000": {},
27+
"cu126": {
28+
"nvidia": {
29+
"cuda_version_lower_bound": [
30+
"12.0"
31+
],
32+
"sm_arch": [
33+
"50_real",
34+
"60_real",
35+
"70_real",
36+
"75_real",
37+
"80_real",
38+
"86_real",
39+
"90_real"
40+
]
41+
}
42+
},
43+
"cu128": {
44+
"nvidia": {
45+
"cuda_version_lower_bound": [
46+
"12.8"
47+
],
48+
"sm_arch": [
49+
"100_real",
50+
"120_real",
51+
"70_real",
52+
"75_real",
53+
"80_real",
54+
"86_real",
55+
"90_real"
56+
]
57+
}
58+
},
59+
"cu129": {
60+
"nvidia": {
61+
"cuda_version_lower_bound": [
62+
"12.9"
63+
],
64+
"sm_arch": [
65+
"100_real",
66+
"120_real",
67+
"120_virtual",
68+
"70_real",
69+
"75_real",
70+
"80_real",
71+
"86_real",
72+
"90_real"
73+
]
74+
}
75+
},
76+
"xpu": {
77+
"intel": {
78+
"device_ip": [
79+
"12.55.8",
80+
"12.60.7",
81+
"12.71.4",
82+
"12.74.4",
83+
"20.1.0",
84+
"20.4.4"
85+
]
86+
}
87+
}
88+
}
89+
}

0 commit comments

Comments
 (0)