Remove some changes.

Author: Xreki

graph_net/subgraph_decompose_and_evaluation_step.py

@@ -396,7 +396,7 @@ def main(args):
 
     failed_decomposition = []
     final_used_splits_map = {}
-    while need_decompose:
+    if need_decompose:
         decomposed_samples_dir = os.path.join(
             pass_work_dir, "samples" if args.framework == "torch" else "paddle_samples"
         )
@@ -405,6 +405,7 @@ def main(args):
         for model_name, task_info in tasks_map.items():
             original_path = task_info["original_path"]
             split_positions = sorted(list(task_info["split_positions"]))
+
             final_used_splits_map[model_name] = split_positions
 
             rectied_model_path = get_rectfied_model_path(original_path)
@@ -429,14 +430,6 @@ def main(args):
         if failed_decomposition:
             print(f"[WARN] {len(failed_decomposition)} models failed to decompose.")
 
-        if not failed_decomposition and num_decomposed_samples == len(tasks_map):
-            need_decompose = True
-            shutil.rmtree(decomposed_samples_dir)
-            os.makedirs(decomposed_samples_dir, exist_ok=True)
-            current_max_size = max(1, current_max_size // 2)
-        else:
-            need_decompose = False
-
     # --- Step 4: Testing ---
     if task_controller.task_scheduler["run_evaluation"]:
         print("\n--- Phase 2: Batch Testing ---")

graph_net/test/subgraph_decompose_and_evaluation_step_test.sh

@@ -1,13 +1,10 @@
 #!/bin/bash
 
-export PYTHONPATH=/work/GraphNet:/work/abstract_pass/Athena:$PYTHONPATH
-
 GRAPH_NET_ROOT=$(python3 -c "import graph_net; import os; print(os.path.dirname(graph_net.__file__))")
 
-FRAMEWORK="paddle"
-LOG_FILE="$GRAPH_NET_ROOT/test/log_xpu.txt"
-#LOG_FILE="/work/GraphNet/benchmark_results/log_test_target_device-xpu_p800_nope-pd_20251111_2.txt"
-OUTPUT_DIR="outputs"
+FRAMEWORK="torch"
+LOG_FILE="$GRAPH_NET_ROOT/test/log_file_for_subgraph_decompose_and_evaluation_step.log"
+OUTPUT_DIR="/tmp/decompose_and_evaluation_workspace"
 TOLERANCE=0
 INITIAL_MAX_SIZE=4096