From e8b0fc7a31e7eb44ca4b9c9f7d94b41e57436da9 Mon Sep 17 00:00:00 2001
From: fangfangssj <1135470306@qq.com>
Date: Sat, 29 Nov 2025 00:41:17 +0800
Subject: [PATCH] fix_test

---
 graph_net/test/decomposer_validator_test.sh | 28 ++++++++++-----------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/graph_net/test/decomposer_validator_test.sh b/graph_net/test/decomposer_validator_test.sh
index b1d3b01d6..4f0099865 100644
--- a/graph_net/test/decomposer_validator_test.sh
+++ b/graph_net/test/decomposer_validator_test.sh
@@ -13,18 +13,22 @@ cp -r "$GRAPH_NET_ROOT/../samples/$MODEL_PATH_IN_SAMPLES" "$OUTPUT_DIR/"
 
 extractor_config_json_str=$(cat <<EOF
 {
-    "custom_extractor_path": "$GRAPH_NET_ROOT/torch/naive_graph_decomposer.py",
-    "custom_extractor_config": {
-        "output_dir": "$OUTPUT_DIR/${MODEL_NAME}_decomposed",
-        "split_positions": [8, 16, 32],
-        "group_head_and_tail": true,
-        "chain_style": true
+    "decorator_path": "$GRAPH_NET_ROOT/torch/extractor.py",
+    "decorator_config": {
+        "name": "$MODEL_NAME",
+        "custom_extractor_path": "$GRAPH_NET_ROOT/torch/naive_graph_decomposer.py",
+        "custom_extractor_config": {
+            "output_dir": "$OUTPUT_DIR/${MODEL_NAME}_decomposed",
+            "split_positions": [8, 16, 32],
+            "group_head_and_tail": true,
+            "chain_style": true
+        }
     }
 }
 EOF
 )
 EXTRACTOR_CONFIG=$(echo $extractor_config_json_str | base64 -w 0)
-python3 -m graph_net.torch.single_device_runner --model-path $GRAPH_NET_ROOT/../samples/$MODEL_PATH_IN_SAMPLES --enable-extract True --extract-name $MODEL_NAME --dump-graph-hash-key --extractor-config=$EXTRACTOR_CONFIG
+python3 -m graph_net.torch.run_model --model-path $GRAPH_NET_ROOT/../samples/$MODEL_PATH_IN_SAMPLES --decorator-config=$EXTRACTOR_CONFIG
 
 FILE_PATH=$GRAPH_NET_DECOMPOSE_PATH/decomposer
 mkdir -p "$(dirname "$FILE_PATH/log.log")"
@@ -35,13 +39,9 @@ python -m graph_net.torch.test_compiler \
     --compiler range_decomposer_validator \
     --device cuda > "$FILE_PATH/log.log" 2>&1
 
-python -m graph_net.log2json \
-    --log-file "$FILE_PATH/log.log" \
-    --output-dir "$FILE_PATH/JSON_results/"
-
-python -m graph_net.plot_ESt \
-    --benchmark-path "$FILE_PATH/JSON_results/" \
-    --output-dir "$FILE_PATH"
+python3 -m graph_net.plot_ESt \
+  --benchmark-path $FILE_PATH/log.log \
+  --output-dir $FILE_PATH \
 
 echo "=================================================="
 echo "Results saved in: $FILE_PATH/ES_result.png"