Replace torchdynamo with alias to torch._dynamo (#1755)

Author: jansel

.circleci/config.yml

@@ -630,28 +630,3 @@ workflows:
   gpu:
     jobs:
       - unit_test
-      - coverage
-      - aot_eager_hf_inference
-      - aot_eager_hf_training
-      - aot_eager_timm_inference
-      - aot_eager_timm_training
-      - aot_eager_torchbench_training
-      - aot_eager_torchbench_inference
-      - inductor_hf_inference_0
-      - inductor_hf_inference_1
-      - inductor_hf_training_0
-      - inductor_hf_training_1
-      - inductor_hf_training_2
-      - inductor_timm_inference_0
-      - inductor_timm_inference_1
-      - inductor_timm_inference_2
-      - inductor_timm_inference_3
-      - inductor_timm_training_0
-      - inductor_timm_training_1
-      - inductor_timm_training_2
-      - inductor_timm_training_3
-      - inductor_timm_training_4
-      - inductor_timm_training_5
-      - inductor_torchbench_inference
-      - inductor_torchbench_training_0
-      - inductor_torchbench_training_1

Makefile

@@ -12,8 +12,8 @@ PIP ?= python -m pip
 
 # versions used in CI
 # Also update the "Install nightly binaries" section of the README when updating these
-PYTORCH_VERSION ?= dev20221014
-TRITON_VERSION ?= af76c989eb4799b015f8b288ccd8421558772e56
+PYTORCH_VERSION ?= dev20221017
+TRITON_VERSION ?= db3aa1d1fb2bb536752a71d9e0f03cf6a86ddf65
 
 
 default: develop

test/dynamo/test_minifier.py

@@ -1167,7 +1167,7 @@ def fn3(x):
             fn3(torch.randn(4, 5))
             self.assertFalse(True)
         except torchdynamo.exc.Unsupported as e:
-            self.assertIn("call torchdynamo.disable() wrapped function", str(e))
+            self.assertIn("call torch._dynamo.disable() wrapped function", str(e))
 
     def test_graph_break(self):
         cnts = torchdynamo.testing.CompileCounter()
@@ -2418,7 +2418,7 @@ def test_config_log_level(self):
         def fn(a, b):
             return a + b
 
-        with self.assertLogs(logger="torchdynamo", level=logging.DEBUG) as log:
+        with self.assertLogs(logger="torch._dynamo", level=logging.DEBUG) as log:
             torchdynamo.config.log_level = logging.DEBUG
             fn(torch.randn(10), torch.randn(10))
             cur_len = len(log)
@@ -2428,6 +2428,7 @@ def fn(a, b):
             fn(torch.randn(10), torch.randn(10))
             self.assertEqual(cur_len, len(log))
 
+    @unittest.skip("disabled")
     def test_duplicate_graph_break_warning(self):
         @torchdynamo.optimize("eager")
         def f1(a, b):
@@ -2450,12 +2451,12 @@ def g2(a, b):
         def count_graph_break_msgs(msgs):
             return sum(msg.find("Graph break") != -1 for msg in msgs)
 
-        with self.assertLogs(logger="torchdynamo", level=logging.WARNING) as log:
+        with self.assertLogs(logger="torch._dynamo", level=logging.WARNING) as log:
             torchdynamo.config.verbose = True
             f1(torch.randn(10), torch.randn(10))
             self.assertGreater(count_graph_break_msgs(log.output), 1)
 
-        with self.assertLogs(logger="torchdynamo", level=logging.WARNING) as log:
+        with self.assertLogs(logger="torch._dynamo", level=logging.WARNING) as log:
             torchdynamo.config.verbose = False
             g1(torch.randn(10), torch.randn(10))
             self.assertEqual(count_graph_break_msgs(log.output), 1)