Fix the config saving of test_target_device.

Author: Xreki

graph_net/subgraph_decompose_and_evaluation_step.py

@@ -447,6 +447,19 @@ def main(args):
         ) = execute_decomposition_phase(
             max_subgraph_size, tasks_map, args.framework, pass_work_dir
         )
+    else:
+        config = load_decompose_config(pass_work_dir)
+        max_subgraph_size = config["max_subgraph_size"]
+        failed_decomposition = config["failed_decomposition_models"]
+        active_models_map = config.get("active_models_map", {})
+        split_positions_map = config.get("split_positions_map", {})
+        for model_name in split_positions_map.keys():
+            tasks_map[model_name] = {
+                "subgraph_path": None,
+                "original_path": active_models_map[model_name],
+                "subgraph_size": None,
+                "split_positions": split_positions_map[model_name],
+            }
 
     # --- Step 4: Testing ---
     pass_log_path = os.path.join(pass_work_dir, "batch_test_result.log")