@@ -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
146146manylinux_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