Update

leslie-fang-intel · leslie-fang-intel · commit 0e955eae0248 · 2025-02-13T19:32:16.000-08:00
[ghstack-poisoned]
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(
diff --git a/torch/_inductor/lowering.py b/torch/_inductor/lowering.py
@@ -1006,29 +1006,15 @@ def squeeze(x, dim=None):
     dims = OrderedSet((dim,) if not isinstance(dim, tuple) else dim)
 
     new_shape = []
-    new_stride = []
-    is_storage_and_layout = ir.is_storage_and_layout(x)
-    original_stride = x.get_stride() if is_storage_and_layout else []
-    new_offset = x.get_layout().offset if is_storage_and_layout else None
     for d, s in enumerate(x.get_size()):
         if not (
             d in dims
             and V.graph.sizevars.evaluate_expr(sympy.Eq(s, 1), size_oblivious=True)
         ):
             new_shape.append(s)
-            if is_storage_and_layout:
-                new_stride.append(original_stride[d])
 
     # squeeze does nothing if the size isn't 1
-    return (
-        (
-            as_strided(x, new_shape, new_stride, new_offset)
-            if is_storage_and_layout
-            else view(x, new_shape)
-        )
-        if new_shape != x.get_size()
-        else x
-    )
+    return view(x, new_shape) if new_shape != x.get_size() else x
 
 
 @register_lowering(aten.squeeze_copy, type_promotion_kind=None)
