move tests

pytorch · Nov 8, 2023 · b5fe64c · b5fe64c
1 parent cecebbe
commit b5fe64c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/test/inductor/test_custom_lowering.py b/test/inductor/test_custom_lowering.py
@@ -137,17 +137,6 @@ def fn(inp, offsets, max_seq_len):
             fn(inp, offsets, max_seq_len), fn_opt(inp, offsets, max_seq_len)
         )
 
-    def test_unfold_zero_dimension_tensor(self):
-        def forward(x):
-            return torch.unfold_copy(dimension=1, input=x,size=0,step=7)
-
-        x = torch.rand([1,0], dtype=torch.float32)
-
-        y = forward(x)
-        compiled_y = torch.compile(forward, mode='max-autotune',fullgraph=True)(x)
-
-        self.assertEqual(y, compiled_y)
-
 
 if __name__ == "__main__":
     from torch._dynamo.test_case import run_tests

diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -6367,6 +6367,17 @@ def test_zero_dim_reductions(self):
 
             self.assertTrue(torch.allclose(actual, expected, atol=1e-3, rtol=1e-3))
 
+    def test_unfold_zero_dimension_tensor(self):
+        def forward(x):
+            return torch.unfold_copy(dimension=1, input=x,size=0,step=7)
+
+        x = torch.rand([1,0], dtype=torch.float32)
+
+        y = forward(x)
+        compiled_y = torch.compile(forward, fullgraph=True)(x)
+
+        self.assertEqual(y, compiled_y)
+
     def test_zero_element_mutation(self):
         class CustomModel(nn.Module):
             def __init__(self):