Skip to content

Commit 85983a5

Browse files
authored
Remove cu116 from circleci (#7183)
1 parent 4ee07cc commit 85983a5

File tree

2 files changed

+12
-184
lines changed

2 files changed

+12
-184
lines changed

.circleci/config.yml

Lines changed: 8 additions & 179 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

.circleci/regenerate.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
3232
for os_type in ["linux", "macos", "win"]:
3333
python_versions = PYTHON_VERSIONS
3434
cu_versions_dict = {
35-
"linux": ["cpu", "cu116", "cu117", "cu118", "rocm5.2", "rocm5.3"],
36-
"win": ["cpu", "cu116", "cu117", "cu118"],
35+
"linux": ["cpu", "cu117", "cu118", "rocm5.2", "rocm5.3"],
36+
"win": ["cpu", "cu117", "cu118"],
3737
"macos": ["cpu"],
3838
}
3939
cu_versions = cu_versions_dict[os_type]
@@ -144,7 +144,6 @@ def upload_doc_job(filter_branch):
144144

145145

146146
manylinux_images = {
147-
"cu116": "pytorch/manylinux-cuda116",
148147
"cu117": "pytorch/manylinux-cuda117",
149148
"cu118": "pytorch/manylinux-cuda118",
150149
}
@@ -271,7 +270,7 @@ def unittest_workflows(indentation=6):
271270
if device_type == "gpu":
272271
if python_version != "3.8":
273272
job["filters"] = gen_filter_branch_tree("main", "nightly")
274-
job["cu_version"] = "cu116"
273+
job["cu_version"] = "cu117"
275274
else:
276275
job["cu_version"] = "cpu"
277276

@@ -289,7 +288,7 @@ def cmake_workflows(indentation=6):
289288
for device in device_types:
290289
job = {"name": f"cmake_{os_type}_{device}", "python_version": python_version}
291290

292-
job["cu_version"] = "cu116" if device == "gpu" else "cpu"
291+
job["cu_version"] = "cu117" if device == "gpu" else "cpu"
293292
if device == "gpu" and os_type == "linux":
294293
job["wheel_docker_image"] = "pytorch/manylinux-cuda116"
295294
jobs.append({f"cmake_{os_type}_{device}": job})

0 commit comments

Comments
 (0)