[SOT] Use tuple InputSpec to avoid type check error (#69853)

SigureMo · web-flow · commit 0f90243955d1 · 2024-12-03T10:50:17.000+08:00
diff --git a/python/paddle/jit/sot/opcode_translator/executor/function_graph.py b/python/paddle/jit/sot/opcode_translator/executor/function_graph.py
@@ -433,7 +433,7 @@ def compile_graph(self, *ret_vars: VariableBase) -> CompileGraphResult:
         symbolic_inputs = self._find_tensor_inputs(input_names)
         compiled_fn = self.sir_ctx.compile_fn(
             statement_ir.name,
-            [var.meta.to_input_spec() for var in symbolic_inputs],
+            tuple(var.meta.to_input_spec() for var in symbolic_inputs),
             **self._kwargs,
         )
         return compiled_fn, (statement_ir, symbolic_inputs, symbolic_outputs)
diff --git a/python/paddle/jit/sot/symbolic/compile_cache.py b/python/paddle/jit/sot/symbolic/compile_cache.py
@@ -104,7 +104,10 @@ def amp_cast_inputs(self, args, kwargs):
     def graph_size(self):
         if self.partial_program is None:
             input_spec = convert_meta_to_input_spec(
-                [self.SIR.symbol_meta_map[symbol] for symbol in self.SIR.inputs]
+                tuple(
+                    self.SIR.symbol_meta_map[symbol]
+                    for symbol in self.SIR.inputs
+                )
             )
             (
                 self.concrete_program,
@@ -181,7 +184,7 @@ def key_fn(
         self,
         context: SymbolicTraceContext,
         sir_name: str,
-        input_spec: list[InputSpec],
+        input_spec: tuple[InputSpec, ...],
         **kwargs,
     ):
         """
@@ -204,7 +207,7 @@ def value_fn(
         self,
         context: SymbolicTraceContext,
         sir_name: str,
-        input_spec: list[InputSpec],
+        input_spec: tuple[InputSpec, ...],
         **kwargs,
     ):
         """
diff --git a/python/paddle/jit/sot/symbolic/symbolic_context.py b/python/paddle/jit/sot/symbolic/symbolic_context.py
@@ -156,7 +156,9 @@ def graph_size(self):
 
         return DummyFunc()
 
-    def compile_fn(self, sir_name: str, input_spec: list[InputSpec], **kwargs):
+    def compile_fn(
+        self, sir_name: str, input_spec: tuple[InputSpec, ...], **kwargs
+    ):
         """
         start compile and return the python function, which must can be to_static without errors.
         """

Original file line number	Diff line number	Diff line change
`@@ -433,7 +433,7 @@ def compile_graph(self, *ret_vars: VariableBase) -> CompileGraphResult:`
`433`	`433`	`symbolic_inputs = self._find_tensor_inputs(input_names)`
`434`	`434`	`compiled_fn = self.sir_ctx.compile_fn(`
`435`	`435`	`statement_ir.name,`
`436`		`- [var.meta.to_input_spec() for var in symbolic_inputs],`
	`436`	`+ tuple(var.meta.to_input_spec() for var in symbolic_inputs),`
`437`	`437`	`**self._kwargs,`
`438`	`438`	`)`
`439`	`439`	`return compiled_fn, (statement_ir, symbolic_inputs, symbolic_outputs)`