add value limit

Author: GoldenStain

python/paddle/jit/sot/opcode_translator/executor/variables/basic.py

@@ -837,7 +837,7 @@ def get_py_value(self, allow_tensor: bool = False) -> bool | int | float:
         self.need_guard_value = True
         log(
             3,
-            f"get_py_value from SymbolicVariable {self} caused value need guard",
+            f"get_py_value from SymbolicVariable {self} caused value need guard\n",
         )
         if isinstance(self.value, SymbolicValue):
             assert isinstance(
@@ -941,6 +941,8 @@ def should_create_symbolic_variable(
         tracker: Tracker,
         symbolic_inputs: dict[str, dict[int, int] | None],
     ):
+        if value < 2:
+            return False
         tracker_expr = tracker.trace_value_from_frame().inlined_expr
         symbolic_inputs.setdefault(tracker_expr, {})
         if tracker_expr in symbolic_inputs:

test/sot/test_sot_dynamic_shape.py

@@ -100,7 +100,7 @@ def test_dynamic_int_input_cache_hit_case1(self):
                 self.assert_results(
                     dynamic_int_input_func1, paddle.randn([3, 4, 5]), i
                 )
-                self.assertEqual(ctx.translate_count, 2)
+                self.assertEqual(ctx.translate_count, 2 if i == 2 else 3)
 
     def test_dynamic_int_input_cache_hit_case2(self):
         with allow_dynamic_shape_guard(
@@ -114,7 +114,7 @@ def test_dynamic_int_input_cache_hit_case2(self):
                 self.assert_results(
                     dynamic_int_input_func2, paddle.randn([3, 4, 5]), {1: i}
                 )
-                self.assertEqual(ctx.translate_count, 2)
+                self.assertEqual(ctx.translate_count, 2 if i == 2 else 3)
 
     def test_dynamic_int_input_cache_hit_case3(self):
         with allow_dynamic_shape_guard(
@@ -138,7 +138,7 @@ def test_dynamic_shape_input_cache_hit_case1(self):
                 self.assert_results(
                     dynamic_shape_input_func1, paddle.randn([i, 4, 5])
                 )
-                self.assertEqual(ctx.translate_count, 2)
+                self.assertEqual(ctx.translate_count, 2 if i == 2 else 3)
 
     def test_dynamic_shape_input_cache_hit_case2(self):
         with allow_dynamic_shape_guard(
@@ -153,7 +153,7 @@ def test_dynamic_shape_input_cache_hit_case2(self):
                     dynamic_shape_access_inner_var_shape,
                     paddle.randn([i, 4, 5]),
                 )
-                self.assertEqual(ctx.translate_count, 2)
+                self.assertEqual(ctx.translate_count, 2 if i == 2 else 3)
 
     def test_dynamic_shape_cast(self):
         with allow_dynamic_shape_guard(
@@ -182,7 +182,7 @@ def test_dynamic_shape_in_list(self):
                     paddle.randn([i, 4, 5]),
                     [i * 4, 5],
                 )
-                self.assertEqual(ctx.translate_count, 2)
+                self.assertEqual(ctx.translate_count, 2 if i == 2 else 3)
 
     def test_conv_dynamic_shape_stride_fallback(self):
         with allow_dynamic_shape_guard(

test/sot/test_trace_list_arg.py

@@ -70,14 +70,14 @@ def test_bar_dynamic_shape(self):
             self.assertEqual(cache.translate_count, 1)
             self.assert_results(bar, a, 2, 0)  # Cache miss
             self.assertEqual(cache.translate_count, 2)
-            self.assert_results(bar, b, 2, 0)  # Cache miss
-            self.assertEqual(cache.translate_count, 3)
             self.assert_results(bar, b, 2, 0)  # Cache hit
-            self.assertEqual(cache.translate_count, 3)
+            self.assertEqual(cache.translate_count, 2)
+            self.assert_results(bar, b, 2, 0)  # Cache hit
+            self.assertEqual(cache.translate_count, 2)
             self.assert_results(bar, b, 1, 1)  # Cache hit
-            self.assertEqual(cache.translate_count, 3)
+            self.assertEqual(cache.translate_count, 2)
             self.assert_results(bar, b, 0, 2)  # Cache miss
-            self.assertEqual(cache.translate_count, 4)
+            self.assertEqual(cache.translate_count, 3)
 
 
 if __name__ == "__main__":