[SOT][DynamicShape] Disallow convert bool to SymbolicVariable, add dynamic axes log (#65477)

SigureMo · web-flow · commit 52315c46f0f3 · 2024-06-27T11:30:43.000+08:00
diff --git a/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py b/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py
@@ -32,6 +32,7 @@
     ConstTypes,
     FallbackError,
     NameGenerator,
+    log,
     paddle_tensor_methods,
     printable,
 )
@@ -330,21 +331,40 @@ def __init__(
         self.meta = meta
         dynamic_axes: list[int] = []
         if ENV_SOT_ALLOW_DYNAMIC_SHAPE.get() and self.tracker.is_traceable():
-            dynamic_axes = self.analyse_dynamic_axes()
+            dynamic_axes = self.analyse_dynamic_axes(tracker)
         self.meta = self.meta.with_dynamic_axes(dynamic_axes)
         self.origin_meta = self.meta
         self.var_name = TensorVariable.var_name_generator.next()
         self.graph.side_effects.record_mutable_variable(self)
 
-    def analyse_dynamic_axes(self):
+    def analyse_dynamic_axes(self, tracker: Tracker):
+        from ..executor_cache import OpcodeExecutorCache
+
         shape_dims = (
             self.shape.proxy.get_all()
         )  # Trigger convert all shape dims to Variable
-        return [
+        dynamic_axes = [
             i
             for i, dim in enumerate(shape_dims)
             if isinstance(dim, SymbolicVariable)
         ]
+        if dynamic_axes:
+            tracker_expr = tracker.trace_value_from_frame().inlined_expr
+            symbolic_inputs = OpcodeExecutorCache().get_symbolic_inputs(
+                self.graph.pycode_gen._origin_code
+            )
+            log(
+                1,
+                f"Start analyse dynamic axes for {tracker.trace_value_from_frame().inlined_expr} in {self.graph.pycode_gen._origin_code}\n",
+            )
+            for key in symbolic_inputs:
+                if key.startswith(tracker_expr):
+                    log(1, f"  {key}: {symbolic_inputs[key]}\n")
+            log(
+                1,
+                f"  -> Tensor {tracker_expr} with dynamic axes {dynamic_axes}\n",
+            )
+        return dynamic_axes
 
     def __len__(self):
         if isinstance(self.meta.shape[0], SymbolicInt):
@@ -742,7 +762,7 @@ def from_value(value: Any, graph: FunctionGraph, tracker: Tracker):
                 tensor_call_shape_var,
                 ConstantVariable.wrap_literal(shape_idx, graph),
             )
-        if not isinstance(value, int):
+        if type(value) is not int:
             return None
         if not tracker.is_traceable():
             return None
