Merge branch 'main' into please_dont_modify_this_branch_unless_you_ar…

…e_just_merging_with_main
pytorch · Jan 9, 2023 · 4b8c4c4 · 4b8c4c4
2 parents 8e9c70e + 35f68a0
commit 4b8c4c4
Show file tree

Hide file tree

Showing 11 changed files with 433 additions and 9 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py
@@ -32,8 +32,8 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
         for os_type in ["linux", "macos", "win"]:
             python_versions = PYTHON_VERSIONS
             cu_versions_dict = {
-                "linux": ["cpu", "cu116", "cu117", "rocm5.2", "rocm5.3"],
-                "win": ["cpu", "cu116", "cu117"],
+                "linux": ["cpu", "cu116", "cu117", "cu118", "rocm5.2", "rocm5.3"],
+                "win": ["cpu", "cu116", "cu117", "cu118"],
                 "macos": ["cpu"],
             }
             cu_versions = cu_versions_dict[os_type]
@@ -145,6 +145,7 @@ def upload_doc_job(filter_branch):
 manylinux_images = {
     "cu116": "pytorch/manylinux-cuda116",
     "cu117": "pytorch/manylinux-cuda117",
+    "cu118": "pytorch/manylinux-cuda118",
 }