[Inductor] Add a JIT Inductor unit test following #146293

desertfire · desertfire · commit 08285df18204 · 2025-02-06T07:04:01.000-08:00
Summary: To follow up #146293, add a JIT Inductor unit test. Other Triton template may need similar fixes. ghstack-source-id: 6fdb192 Pull Request resolved: #146529
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -3747,6 +3747,26 @@ def forward(self, x):
         with self.assertRaisesRegex(RuntimeError, msg):
             torch.compile(fn)(t)
 
+    @config.patch(
+        {
+            "max_autotune": True,
+            "max_autotune_gemm_backends": "TRITON",
+        }
+    )
+    def test_linear_dynamic_maxautotune(self):
+        @torch.compile(dynamic=True)
+        class Model(torch.nn.Module):
+            def __init__(self) -> None:
+                super().__init__()
+                self.linear = torch.nn.Linear(1, 1)
+
+            def forward(self, x):
+                return self.linear(x)
+
+        x = torch.randn(10, 1)
+        torch._dynamo.mark_dynamic(x, 0)
+        self.common(Model(), (x,))
+
     def test_scalar_input(self):
         def fn(x, y):
             a = torch.div(x, y, rounding_mode="floor")
