Merge remote-tracking branch 'origin/master' into HEAD

.circleci/cimodel/data/caffe2_build_data.py

@@ -11,7 +11,6 @@
         (Ver("gcc", "4.9"), [X("py2")]),
     ]),
     (Ver("ubuntu", "16.04"), [
-        (Ver("cuda", "8.0"), [X("py2")]),
         (Ver("cuda", "9.0"), [
             # TODO make explicit that this is a "secret TensorRT build"
             #  (see https://github.com/pytorch/pytorch/pull/17323#discussion_r259446749)

.circleci/cimodel/data/caffe2_build_definitions.py

@@ -10,7 +10,7 @@
 
 DOCKER_IMAGE_PATH_BASE = "308535385114.dkr.ecr.us-east-1.amazonaws.com/caffe2/"
 
-DOCKER_IMAGE_VERSION = 273
+DOCKER_IMAGE_VERSION = 276
 
 
 class Conf(object):

.circleci/cimodel/data/pytorch_build_data.py

@@ -8,7 +8,7 @@
         (None, [
             X("2.7.9"),
             X("2.7"),
-            X("3.5"),
+            ("3.5", [("important", [X(True)])]),
             X("nightly"),
         ]),
         ("gcc", [
@@ -28,7 +28,6 @@
             ("5", [X("3.6")]),
         ]),
         ("cuda", [
-            ("8", [X("3.6")]),
             ("9", [
                 # Note there are magic strings here
                 # https://github.com/pytorch/pytorch/blob/master/.jenkins/pytorch/build.sh#L21
@@ -37,7 +36,7 @@
                 # and
                 # https://github.com/pytorch/pytorch/blob/master/.jenkins/pytorch/build.sh#L153
                 # (from https://github.com/pytorch/pytorch/pull/17323#discussion_r259453144)
-                X("2.7"),
+                ("2.7", [("important", [X(True)])]),
                 X("3.6"),
             ]),
             ("9.2", [X("3.6")]),
@@ -136,6 +135,7 @@ def child_constructor(self):
         next_nodes = {
             "xla": XlaConfigNode,
             "namedtensor": NamedTensorConfigNode,
+            "important": ImportantConfigNode,
         }
         return next_nodes[experimental_feature]
 
@@ -156,6 +156,14 @@ def init2(self, node_name):
         self.props["is_namedtensor"] = node_name
 
 
+class ImportantConfigNode(TreeConfigNode):
+    def modify_label(self, label):
+        return "IMPORTANT=" + str(label)
+
+    def init2(self, node_name):
+        self.props["is_important"] = node_name
+
+
 class XenialCompilerConfigNode(TreeConfigNode):
 
     def init2(self, node_name):

.circleci/cimodel/data/pytorch_build_definitions.py

@@ -25,7 +25,8 @@ def __init__(self,
                  gpu_resource=None,
                  dependent_tests=None,
                  parent_build=None,
-                 is_namedtensor=False):
+                 is_namedtensor=False,
+                 is_important=False):
 
         self.distro = distro
         self.pyver = pyver
@@ -37,6 +38,7 @@ def __init__(self,
         # (from https://github.com/pytorch/pytorch/pull/17323#discussion_r259453608)
         self.is_xla = is_xla
         self.is_namedtensor = is_namedtensor
+        self.is_important = is_important
 
         self.restrict_phases = restrict_phases
         self.gpu_resource = gpu_resource
@@ -46,7 +48,10 @@ def __init__(self,
     # TODO: Eliminate the special casing for docker paths
     # In the short term, we *will* need to support special casing as docker images are merged for caffe2 and pytorch
     def get_parms(self, for_docker):
-        leading = ["pytorch"]
+        leading = []
+        if self.is_important and not for_docker:
+            leading.append("AAA")
+        leading.append("pytorch")
         if self.is_xla and not for_docker:
             leading.append("xla")
         if self.is_namedtensor and not for_docker:
@@ -225,6 +230,7 @@ def instantiate_configs():
 
         is_xla = fc.find_prop("is_xla") or False
         is_namedtensor = fc.find_prop("is_namedtensor") or False
+        is_important = fc.find_prop("is_important") or False
 
         gpu_resource = None
         if cuda_version and cuda_version != "10":
@@ -239,9 +245,10 @@ def instantiate_configs():
             restrict_phases,
             gpu_resource,
             is_namedtensor=is_namedtensor,
+            is_important=is_important,
         )
 
-        if cuda_version == "8":
+        if cuda_version == "9" and python_version == "3.6":
             c.dependent_tests = gen_dependent_configs(c)
 
         config_list.append(c)