Update

[ghstack-poisoned]
pytorch · XuehaiPan · Oct 5, 2024 · Oct 5, 2024 · Oct 5, 2024 · Oct 5, 2024
commit 1d1a757bfeb6ed529632b67449f2815d0318bee6
diff --git a/test/dynamo/test_misc.py b/test/dynamo/test_misc.py
@@ -8683,7 +8683,7 @@ def fn():
 
     def test_tracing_pytree(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(xs):
                     flat_xs, spec = module.tree_flatten(xs)
@@ -8699,7 +8699,7 @@ def fn(xs):
 
     def test_tracing_nested_pytree(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(xs):
                     flat_xs, spec = module.tree_flatten(xs)
@@ -8718,7 +8718,7 @@ def fn(xs):
 
     def test_tracing_nested_pytree_tuples(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(xs):
                     flat_xs, spec = module.tree_flatten(xs)
@@ -8737,7 +8737,7 @@ def fn(xs):
 
     def test_tracing_nested_pytree_dicts(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(xs):
                     flat_xs, spec = module.tree_flatten(xs)
@@ -8776,7 +8776,7 @@ def fn(x):
 
     def test_tracing_nested_pytree_mixed_all(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(xs):
                     flat_xs, spec = module.tree_flatten(xs)
@@ -8830,7 +8830,7 @@ def test_tracing_py_tree_tensor_subclass(self):
         from torch.utils.checkpoint import checkpoint
 
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(xs):
                     nested_xs = [[xs]]
@@ -8850,7 +8850,7 @@ def checkpoint_fn(xs):
 
     def test_tracing_tree_map_only(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(xs):
                     def mapper(x):
@@ -10219,7 +10219,7 @@ def fn(x, y):
 
     def test_pytree_tree_leaves(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(x):
                     tree = {
@@ -10249,7 +10249,7 @@ def fn(x):
 
     def test_pytree_tree_flatten_unflatten(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(x, y):
                     tree = {
@@ -10296,7 +10296,7 @@ def fn(x, y):
 
     def test_pytree_tree_map(self):
         for name, module in pytree_modules.items():
-            with self.subTest(f"pytree implement: {name}"):
+            with self.subTest(f"pytree implementation: {name}"):
 
                 def fn(x, y):
                     tree1 = {