@@ -32,8 +32,8 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
32
32
for os_type in ["linux" , "macos" , "win" ]:
33
33
python_versions = PYTHON_VERSIONS
34
34
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" ],
37
37
"macos" : ["cpu" ],
38
38
}
39
39
cu_versions = cu_versions_dict [os_type ]
@@ -144,7 +144,6 @@ def upload_doc_job(filter_branch):
144
144
145
145
146
146
manylinux_images = {
147
- "cu116" : "pytorch/manylinux-cuda116" ,
148
147
"cu117" : "pytorch/manylinux-cuda117" ,
149
148
"cu118" : "pytorch/manylinux-cuda118" ,
150
149
}
@@ -271,7 +270,7 @@ def unittest_workflows(indentation=6):
271
270
if device_type == "gpu" :
272
271
if python_version != "3.8" :
273
272
job ["filters" ] = gen_filter_branch_tree ("main" , "nightly" )
274
- job ["cu_version" ] = "cu116 "
273
+ job ["cu_version" ] = "cu117 "
275
274
else :
276
275
job ["cu_version" ] = "cpu"
277
276
@@ -289,7 +288,7 @@ def cmake_workflows(indentation=6):
289
288
for device in device_types :
290
289
job = {"name" : f"cmake_{ os_type } _{ device } " , "python_version" : python_version }
291
290
292
- job ["cu_version" ] = "cu116 " if device == "gpu" else "cpu"
291
+ job ["cu_version" ] = "cu117 " if device == "gpu" else "cpu"
293
292
if device == "gpu" and os_type == "linux" :
294
293
job ["wheel_docker_image" ] = "pytorch/manylinux-cuda116"
295
294
jobs .append ({f"cmake_{ os_type } _{ device } " : job })
0 commit comments