【save_load】fix Jit load func_name bug (#71457)

xiaoguoguo626807 · web-flow · commit c9d0c30dbb8b · 2025-03-07T10:07:38.000+08:00
* add customdevice default_pass

* add declare

* rm stdmove

* fix jit load

* fix jit load

* fix error
diff --git a/python/paddle/jit/api.py b/python/paddle/jit/api.py
@@ -751,14 +751,14 @@ def _build_load_path_and_config(path, config):
     directory_format_exist = os.path.isdir(path)
     if prefix_format_exist and directory_format_exist:
         raise ValueError(
-            f"The {path}.pdmodel and {path} directory exist at the same time, "
+            f"The {path}.pdmodel(json) and {path} directory exist at the same time, "
             "don't know which one to load, please make sure that the specified target "
             "of ``path`` is unique."
         )
     elif not prefix_format_exist and not directory_format_exist:
         raise ValueError(
             f"The ``path`` ({path}) to load model not exists. "
-            "Please make sure that *.pdmodel exists or "
+            "Please make sure that *.pdmodel(json) exists or "
             "don't using ``skip_forward=True`` to jit.save."
         )
     else:
diff --git a/python/paddle/jit/pir_translated_layer.py b/python/paddle/jit/pir_translated_layer.py
@@ -279,7 +279,7 @@ def _construct_program_holders(model_path, model_filename=None):
                     else:
                         method_name.replace('model', '')
                     program, trainable = _load_pir_program(model_file_path)
-                    program_holder_dict[func_name] = _PirProgramHolder(
+                    program_holder_dict[method_name] = _PirProgramHolder(
                         program, trainable
                     )
 

Original file line number	Diff line number	Diff line change
`@@ -279,7 +279,7 @@ def _construct_program_holders(model_path, model_filename=None):`
`279`	`279`	`else:`
`280`	`280`	`method_name.replace('model', '')`
`281`	`281`	`program, trainable = _load_pir_program(model_file_path)`
`282`		`- program_holder_dict[func_name] = _PirProgramHolder(`
	`282`	`+ program_holder_dict[method_name] = _PirProgramHolder(`
`283`	`283`	`program, trainable`
`284`	`284`	`)`
`285`	`285`