[Inductor] Fix the lowering of squeeze when input is not contiguous

leslie-fang-intel · leslie-fang-intel · commit d65c85352405 · 2025-02-13T19:32:16.000-08:00
ghstack-source-id: dd5a327 Pull Request resolved: #146746
diff --git a/test/inductor/test_unbacked_symints.py b/test/inductor/test_unbacked_symints.py
@@ -385,9 +385,6 @@ def fn(t, start, length):
     @skipGPUIf(not HAS_GPU, "requires gpu and triton")
     @dynamo_config.patch(capture_dynamic_output_shape_ops=True)
     def test_issue_143498(self, device):
-        if device == "cpu":
-            raise unittest.SkipTest("CPU Failure")
-
         class Model(torch.nn.Module):
             def __init__(self) -> None:
                 super().__init__()
diff --git a/torch/_inductor/ir.py b/torch/_inductor/ir.py
@@ -2769,7 +2769,9 @@ def fake_reindex(index):  # type: ignore[no-untyped-def]
                 # TODO: unbacked should not diverge from backed in determining striding
                 # Need to require contiguous here instead of realize, see:
                 # https://github.com/pytorch/pytorch/issues/145561
-                x = ExternKernel.require_contiguous(x)
+                x = ExternKernel.require_exact_strides(
+                    x, FlexibleLayout.contiguous_strides(x.get_size())
+                )
 
             storage, old_layout = as_storage_and_layout(x, want_contiguous=True)
             new_layout = FixedLayout(
