Skip to content

Commit 96aae0a

Browse files
TroyGardenfacebook-github-bot
authored andcommitted
update matrix filter to exclude cu129 which fbgemm currently doesn't support (pytorch#3103)
Summary: Pull Request resolved: pytorch#3103 # context * pytorch infra bumps the cuda version to have cu129 appears in the matrix * currently fbgemm only supports up to cu128, which causes cu129 build linux wheels failed on github {F1979407457} * this diff update the filter.py to exclude cu129 Differential Revision: D76823583
1 parent bd9e187 commit 96aae0a

File tree

1 file changed

+40
-1
lines changed

1 file changed

+40
-1
lines changed

.github/scripts/filter.py

Lines changed: 40 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,50 @@ def main():
1818

1919
full_matrix_string = os.environ["MAT"]
2020
full_matrix = json.loads(full_matrix_string)
21+
"""
22+
Matrix contents can be found in a github "Build Linux Wheels"
23+
log output. A typical example is
24+
{
25+
"include": [
26+
{
27+
"python_version": "3.9",
28+
"gpu_arch_type": "cpu",
29+
"gpu_arch_version": "",
30+
"desired_cuda": "cpu",
31+
"container_image": "pytorch/manylinux2_28-builder:cpu",
32+
"package_type": "manywheel",
33+
"build_name": "manywheel-py3_9-cpu",
34+
"validation_runner": "linux.2xlarge",
35+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
36+
"channel": "nightly",
37+
"upload_to_base_bucket": "no",
38+
"stable_version": "2.7.1",
39+
"use_split_build": false
40+
},
41+
{
42+
"python_version": "3.9",
43+
"gpu_arch_type": "cuda",
44+
"gpu_arch_version": "12.6",
45+
"desired_cuda": "cu126",
46+
"container_image": "pytorch/manylinux2_28-builder:cuda12.6",
47+
"package_type": "manywheel",
48+
"build_name": "manywheel-py3_9-cuda12_6",
49+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
50+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126",
51+
"channel": "nightly",
52+
"upload_to_base_bucket": "no",
53+
"stable_version": "2.7.1",
54+
"use_split_build": false
55+
}
56+
]
57+
}
58+
"""
2159

2260
new_matrix_entries = []
2361

2462
for entry in full_matrix["include"]:
25-
new_matrix_entries.append(entry)
63+
if entry["desired_cuda"] != "cu129":
64+
new_matrix_entries.append(entry)
2665

2766
new_matrix = {"include": new_matrix_entries}
2867
print(json.dumps(new_matrix))

0 commit comments

Comments
 (0)