Aot Autograd - Dont fallback with new op

Author: anijain2305

benchmarks/common.py

@@ -68,7 +68,6 @@ def set_model_name(name):
     "pytorch_struct",
     "speech_transformer",
     "vision_maskrcnn",
-    "moco",
     # Huggingface
     "AlbertForMaskedLM",  # OOM
     "AlbertForQuestionAnswering",  # OOM

torchdynamo/optimizations/training.py

@@ -34,8 +34,7 @@ def is_aot_autograd_safe_to_run(gm, example_inputs):
     Issues
     1) LSTM - https://github.com/pytorch/torchdynamo/issues/1147
     2) LSTM - https://github.com/pytorch/functorch/issues/586
-    3) New op - https://github.com/pytorch/torchdynamo/issues/1448
-    4) Input mutation - https://github.com/pytorch/torchdynamo/issues/1301
+    3) Input mutation - https://github.com/pytorch/torchdynamo/issues/1301
     """
 
     def raise_or_warn(reason):
@@ -53,11 +52,6 @@ def raise_or_warn(reason):
         if submod.__class__.__name__ == "LSTM":
             return raise_or_warn("LSTM")
 
-    # 2) new does not work with fake tensor and aot autograd
-    for node in gm.graph.nodes:
-        if node.op == "call_method" and node.target == "new":
-            return raise_or_warn("new operator")
-
     # 2) Mutation in the graph
     mutated = False
     try: