[SOT] Add dynamic shape check && rm useless info (#73287)

DrRyanHuang · web-flow · commit f5ddd6ffe691 · 2025-06-13T19:04:15.000+08:00
diff --git a/python/paddle/jit/sot/infer_meta.py b/python/paddle/jit/sot/infer_meta.py
@@ -115,8 +115,7 @@ def unwrap_or_breakgraph(self):
         return self.meta
 
     def unwrap_unsafe(self):
-        if self.meta is None:
-            raise AssertionError("MetaInfo is None")
+        assert self.meta is not None, "MetaInfo is None"
         return self.meta
 
     def with_dynamic_axes(
@@ -222,6 +221,13 @@ def from_value(value) -> MetaInfoOrNull:
         name = SOT_INFER_META_INNER_VAR
         dtype = MetaInfoOrNull._handle_legacy_ir_amp_dtype(value.dtype)
         shape = [SymbolicInt() if dim == -1 else dim for dim in value.shape]
+        for dim in shape:
+            if isinstance(dim, int):
+                assert dim >= 0, (
+                    "Dimensions must be non-negative integers or SymbolicInt. "
+                    f"Encountered value {dim} in shape {shape}."
+                )
+
         if isinstance(value, paddle.pir.Value) and value.is_dist():
             dist_info = DistInfo.from_value(value)
         else:
diff --git a/python/paddle/jit/sot/opcode_translator/executor/function_graph.py b/python/paddle/jit/sot/opcode_translator/executor/function_graph.py
@@ -553,7 +553,7 @@ def call_paddle_api(
         log(3, f"call paddle.api : {func.__name__}", "\n")
 
         def message_handler(*args, **kwargs):
-            return f"Call paddle_api error: {func.__name__}, may be not a operator api?"
+            return f"Call paddle_api error: {func.__name__}"
 
         return inner_error_default_handler(self.symbolic_call, message_handler)(
             InferMetaCache(),
@@ -580,7 +580,7 @@ def call_numpy_api(
         log(3, f"call numpy.api : {func.__name__}", "\n")
 
         def message_handler(*args, **kwargs):
-            return f"Call numpy api error: {func.__name__}, may be not a operator api?"
+            return f"Call numpy api error: {func.__name__}"
 
         return inner_error_default_handler(self.symbolic_call, message_handler)(
             InferMetaCache(),
@@ -623,7 +623,7 @@ def call_tensor_method(
         """
 
         def message_handler(*args, **kwargs):
-            return f"Call tensor_method error: Tensor.{method_name}, may be not a valid operator api?"
+            return f"Call tensor_method error: Tensor.{method_name}"
 
         return inner_error_default_handler(self.symbolic_call, message_handler)(
             InferMetaCache(),
@@ -661,7 +661,7 @@ def compute_fn(layer, inputs, outputs, stacks):
             )
 
         def message_handler(*args, **kwargs):
-            return f"Call paddle layer error: {layer}, may be not a valid paddle layer?"
+            return f"Call paddle layer error: {layer}"
 
         return inner_error_default_handler(self.symbolic_call, message_handler)(
             infer_meta_fn, compute_fn, layer, APIType.PADDLE, *args, **kwargs
