Merge branch 'main' into add-regnet-128-swag

pytorch · Apr 4, 2022 · 5560105 · 5560105
2 parents 278f0f7 + 3130b45
commit 5560105
Show file tree

Hide file tree

Showing 5 changed files with 221 additions and 86 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py
@@ -32,7 +32,7 @@ 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", "cu102", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
+                "linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.5.2", "rocm5.0"],
                 "win": ["cpu", "cu113", "cu115"],
                 "macos": ["cpu"],
             }