change the key of split_results.json from model_name to model_path

Author: JewelRoam

graph_net/torch/graph_decomposer.py

@@ -206,9 +206,7 @@ def _make_config(
     def __call__(self, rel_model_path):
         model_path = os.path.join(self.config["model_path_prefix"], rel_model_path)
         split_results = load_json(self.config["split_results_path"])
-        split_positions = split_results[os.path.basename(rel_model_path)][
-            "split_points"
-        ]
+        split_positions = split_results[rel_model_path]["split_positions"]
         config = {
             "split_positions": split_positions,
             "group_head_and_tail": self.config.get("group_head_and_tail", False),

graph_net/torch/typical_sequence_split_points.py

@@ -203,7 +203,7 @@ def analyze(self, model_paths_file: str, device: str) -> Dict[str, Dict]:
                 )
 
             current_idx = 0
-            split_points_set = set()
+            split_positions = set()
             total_len = sum(token2len.get(t, 1) for t in seq_tokens)
 
             for token_id in seq_tokens:
@@ -212,22 +212,22 @@ def analyze(self, model_paths_file: str, device: str) -> Dict[str, Dict]:
 
                 if is_pattern:
                     if current_idx > 0:
-                        split_points_set.add(current_idx)
+                        split_positions.add(current_idx)
                     end_idx = current_idx + length
                     if end_idx < total_len:
-                        split_points_set.add(end_idx)
+                        split_positions.add(end_idx)
 
                 current_idx += length
 
-            sorted_splits = sorted(list(split_points_set))
+            sorted_splits = sorted(list(split_positions))
 
             self._print_analysis(
                 model_name, str(original_path), sorted_splits, total_len, full_model_ops
             )
 
-            results[model_name] = {
-                "path": str(original_path),
-                "split_points": sorted_splits,
+            results[str(original_path)] = {
+                "model_name": model_name,
+                "split_positions": sorted_splits,
                 "total_length": total_len,
             }