Add extra_mount alias resolution and use generic Eagle model path

yeswanthk-26 · yeswanthk-26 · commit 421742999747 · 2026-04-10T05:18:21.000Z
Add model_paths alias resolution for extra_mount host paths in config.py,
enabling MTP recipes to use generic name "kimi-k2.5-eagle3" instead of
cluster-specific path for the Eagle speculative decoding model.
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen1dep16_batch32_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen1dep16_batch32_eplb0_mtp3.yaml
@@ -115,7 +115,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen1dep32_batch16_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen1dep32_batch16_eplb0_mtp3.yaml
@@ -113,7 +113,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen1dep8_batch512_eplb0_mtp1.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen1dep8_batch512_eplb0_mtp1.yaml
@@ -175,7 +175,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen4tep8_batch64_allconc_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen4tep8_batch64_allconc_eplb0_mtp3.yaml
@@ -120,7 +120,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen5tep4_batch2_allconc_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx1dep4_gen5tep4_batch2_allconc_eplb0_mtp3.yaml
@@ -111,7 +111,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx2dep4_gen1dep16_batch128_eplb0_mtp1.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx2dep4_gen1dep16_batch128_eplb0_mtp1.yaml
@@ -127,7 +127,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx2dep4_gen1dep32_batch64_eplb0_mtp1.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx2dep4_gen1dep32_batch64_eplb0_mtp1.yaml
@@ -119,7 +119,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx2dep4_gen3dep8_batch256_eplb0_mtp1.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL1K_OSL1K/MTP/ctx2dep4_gen3dep8_batch256_eplb0_mtp1.yaml
@@ -143,7 +143,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx1dep4_gen2tep8_batch32_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx1dep4_gen2tep8_batch32_eplb0_mtp3.yaml
@@ -116,7 +116,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx1dep4_gen4tep8_batch1_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx1dep4_gen4tep8_batch1_eplb0_mtp3.yaml
@@ -112,7 +112,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx1dep4_gen5tep4_batch8_allconc_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx1dep4_gen5tep4_batch8_allconc_eplb0_mtp3.yaml
@@ -112,7 +112,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx2dep4_gen1dep16_batch8_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx2dep4_gen1dep16_batch8_eplb0_mtp3.yaml
@@ -112,7 +112,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx5dep4_gen1dep32_batch16_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx5dep4_gen1dep32_batch16_eplb0_mtp3.yaml
@@ -113,7 +113,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx5dep4_gen1dep8_batch256_allconc_eplb0_mtp1.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx5dep4_gen1dep8_batch256_allconc_eplb0_mtp1.yaml
@@ -143,7 +143,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx8dep4_gen1dep32_batch32_eplb0_mtp3.yaml b/recipes/kimi2.5/trtllm_dynamo/disagg/gb200Nvfp4/ISL8K_OSL1K/MTP/ctx8dep4_gen1dep32_batch32_eplb0_mtp3.yaml
@@ -115,7 +115,7 @@ backend:
         speculative_model_dir: "/eagle-model"
 
 extra_mount:
-  - "/mnt/lustre01/users/slurm-shared/yeswanthk/models/Kimi-K2.5-Thinking-Eagle3:/eagle-model"
+  - "kimi-k2.5-eagle3:/eagle-model"
 
 benchmark:
   type: "sa-bench"
diff --git a/src/srtctl/core/config.py b/src/srtctl/core/config.py
@@ -141,6 +141,20 @@ def resolve_config_with_defaults(user_config: dict[str, Any], cluster_config: di
         config["reporting"] = cluster_config["reporting"]
         logger.debug("Applied cluster reporting config")
 
+    # Resolve extra_mount host path aliases through model_paths
+    extra_mounts = config.get("extra_mount", [])
+    if model_paths and extra_mounts:
+        resolved_mounts = []
+        for mount_spec in extra_mounts:
+            host_path, container_path = mount_spec.split(":", 1)
+            if host_path in model_paths:
+                resolved_host = model_paths[host_path]
+                resolved_mounts.append(f"{resolved_host}:{container_path}")
+                logger.debug(f"Resolved extra_mount alias '{host_path}' -> '{resolved_host}'")
+            else:
+                resolved_mounts.append(mount_spec)
+        config["extra_mount"] = resolved_mounts
+
     # Resolve frontend nginx_container alias
     frontend = config.get("frontend", {})
     nginx_container = frontend.get("nginx_container", "")
