[SOT][DynamicShape] Search symbolic inputs by use dict key (#66746)

SigureMo · web-flow · commit 25cae186ace5 · 2024-07-30T17:22:27.000+08:00
diff --git a/python/paddle/jit/sot/opcode_translator/executor/tracker.py b/python/paddle/jit/sot/opcode_translator/executor/tracker.py
@@ -69,18 +69,6 @@ def trace_value_from_frame(self) -> StringifiedExpression:
         """
         raise NotImplementedError()
 
-    def match_expr(self, expr: str) -> bool:
-        """
-        Match the expression with the tracked variables.
-
-        Args:
-            expr (str): The expression to be matched.
-
-        Returns:
-            bool: True if the expression matches the tracked variables, False otherwise.
-        """
-        return self.trace_value_from_frame().inlined_expr == expr
-
     def is_traceable(self) -> bool:
         """
         Determine if all the tracked variables can be traced from the frame.
diff --git a/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py b/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py
@@ -756,15 +756,15 @@ def should_create_symbolic_variable(
     ):
         tracker_expr = tracker.trace_value_from_frame().inlined_expr
         symbolic_inputs.setdefault(tracker_expr, {})
-        for expr, symbolic_input in symbolic_inputs.items():
-            if tracker.match_expr(expr):
-                symbolic_input.setdefault(value, 0)
-                symbolic_input[value] += 1
-                if symbolic_input[value] >= STATIC_DIM_FREQ_THRESHOLD:
-                    return False
-                if len(symbolic_input.keys()) > 1:
-                    return True
+        if tracker_expr in symbolic_inputs:
+            symbolic_input = symbolic_inputs[tracker_expr]
+            symbolic_input.setdefault(value, 0)
+            symbolic_input[value] += 1
+            if symbolic_input[value] >= STATIC_DIM_FREQ_THRESHOLD:
                 return False
+            if len(symbolic_input.keys()) > 1:
+                return True
+            return False
         return False
 
     @staticmethod
