Update

[ghstack-poisoned]
pytorch · Xia-Weiwen · Feb 8, 2025 · Feb 8, 2025 · Feb 9, 2025 · Feb 10, 2025
commit 1e3b6b0c50d357420699cb302743ae60683a7689
diff --git a/torch/_inductor/quantized_lowerings.py b/torch/_inductor/quantized_lowerings.py
@@ -160,6 +160,7 @@ def int4pack_mm_cpu(
         # define functions to generate example inputs for weight and group size
         # otherwise, autotuner generates example inputs of all zeros for them
         def get_example_weight(x: torch._inductor.ir.IRNode) -> torch.Tensor:
+            assert x.get_layout().is_contiguous()
             shape = x.get_size()
             device = x.get_device()
             return torch.randint(0, 255, shape, dtype=torch.uint8, device=device)