Update on "multi_margin_loss: check weight shape, make contiguous…

… on CPU, add tests"

[ghstack-poisoned]

.ci/docker/requirements-ci.txt

@@ -75,10 +75,10 @@ librosa>=0.6.2 ; python_version < "3.11"
 #Pinned versions:
 #test that import:
 
-mypy==0.960
+mypy==1.4.1
 # Pin MyPy version because new errors are likely to appear with each release
 #Description: linter
-#Pinned versions: 0.960
+#Pinned versions: 1.4.1
 #test that import: test_typing.py, test_type_hints.py
 
 networkx==2.8.8

.github/ci_commit_pins/vision.txt

@@ -1 +1 @@
-08c9938f3e69b4fb2d710a038479264f3c23b133
+bb3aae7b2543637191ad9c810f082eae622534b8

.github/labeler.yml

@@ -26,6 +26,8 @@
 - torch/_functorch/aot_autograd.py
 - .ci/docker/ci_commit_pins/**
 - .github/ci_commit_pins/**
+- c10/core/Sym*
+- torch/fx/experimental/symbolic_shapes.py
 
 "module: cpu":
 - aten/src/ATen/cpu/**

.github/scripts/github_utils.py

@@ -2,6 +2,7 @@
 
 import json
 import os
+import warnings
 
 from dataclasses import dataclass
 from typing import Any, Callable, cast, Dict, List, Optional, Tuple
@@ -143,3 +144,26 @@ def gh_post_commit_comment(
 def gh_delete_comment(org: str, repo: str, comment_id: int) -> None:
     url = f"https://api.github.com/repos/{org}/{repo}/issues/comments/{comment_id}"
     gh_fetch_url(url, method="DELETE")
+
+
+def gh_fetch_merge_base(org: str, repo: str, base: str, head: str) -> str:
+    merge_base = ""
+    # Get the merge base using the GitHub REST API. This is the same as using
+    # git merge-base without the need to have git. The API doc can be found at
+    # https://docs.github.com/en/rest/commits/commits?apiVersion=2022-11-28#compare-two-commits
+    try:
+        json_data = gh_fetch_url(
+            f"https://api.github.com/repos/{org}/{repo}/compare/{base}...{head}",
+            headers={"Accept": "application/vnd.github.v3+json"},
+            reader=json.load,
+        )
+        if json_data:
+            merge_base = json_data.get("merge_base_commit", {}).get("sha", "")
+        else:
+            warnings.warn(
+                f"Failed to get merge base for {base}...{head}: Empty response"
+            )
+    except Exception as error:
+        warnings.warn(f"Failed to get merge base for {base}...{head}: {error}")
+
+    return merge_base