fix test

pianpwk · pianpwk · commit 40303c5dac61 · 2025-03-05T13:38:23.000-08:00
diff --git a/test/inductor/test_aot_inductor.py b/test/inductor/test_aot_inductor.py
@@ -4129,7 +4129,7 @@ def forward(self, a, b, c):
         so_path, code = run_and_get_cpp_code(
             AOTIRunnerUtil.compile, model, example_inputs
         )
-        lowerbound_check = "u1 >= 1" if mark_unbacked else "u0 >= 2"
+        lowerbound_check = "u1 >= 1" if mark_unbacked else "u0 >= 1"
         FileCheck().check_count(lowerbound_check, 1).run(code)
 
         compiled = AOTIRunnerUtil.load(self.device, so_path)
diff --git a/test/test_dynamic_shapes.py b/test/test_dynamic_shapes.py
@@ -432,11 +432,12 @@ def test_size_expressions(self):
         else:
             result = expand_x + expand_x
 
-        gt_op, _bt = shape_env.guards[-1]
+        gt_op, _bt, is_size_obv = shape_env.guards[-1]
         self.assertTrue(isinstance(gt_op, sympy.core.relational.StrictGreaterThan))
         self.assertTrue(str(x.shape[0]), str(gt_op.args[0]))
         self.assertTrue(str(expand_x.shape[1]), str(x.shape[0]))
         self.assertTrue(str(expand_x.shape[1]), str(result.shape[0]))
+        self.assertFalse(is_size_obv)
 
     def test_floordiv_static(self):
         shape_env = ShapeEnv()

Original file line number	Diff line number	Diff line change
`@@ -4129,7 +4129,7 @@ def forward(self, a, b, c):`
`4129`	`4129`	`so_path, code = run_and_get_cpp_code(`
`4130`	`4130`	`AOTIRunnerUtil.compile, model, example_inputs`
`4131`	`4131`	`)`
`4132`		`- lowerbound_check = "u1 >= 1" if mark_unbacked else "u0 >= 2"`
	`4132`	`+ lowerbound_check = "u1 >= 1" if mark_unbacked else "u0 >= 1"`
`4133`	`4133`	`FileCheck().check_count(lowerbound_check, 1).run(code)`
`4134`	`4134`
`4135`	`4135`	`compiled = AOTIRunnerUtil.load(self.device, so_path)`