Fix input args

Signed-off-by: jthomson04 <[email protected]>

Author: jthomson04

components/backends/trtllm/performance_sweeps/submit_disagg.sh

@@ -373,8 +373,8 @@ main() {
             fi
             ;;
         "tep")
-            if [ $# -ne 15 ]; then
-                echo "Error: TEP mode requires 15 additional parameters (including mtp_mode)"
+            if [ $# -ne 14 ]; then
+                echo "Error: TEP mode requires 14 additional parameters (including mtp_mode)"
                 usage
             fi
 
@@ -384,22 +384,21 @@ main() {
             local ctx_enable_attention_dp=$6
             local gen_num=$7
             local gen_tp_size=$8
-            local gen_ep_size=$9
-            local gen_batch_size=${10}
-            local gen_max_num_tokens=${11}
-            local gen_gpu_memory_fraction=${12}
-            local gen_mtp_size=${13}
-            local gen_eplb_num_slots=${14}
-            local gen_concurrency_list=${15}
+            local gen_batch_size=$9
+            local gen_max_num_tokens=${10}
+            local gen_gpu_memory_fraction=${11}
+            local gen_mtp_size=${12}
+            local gen_eplb_num_slots=${13}
+            local gen_concurrency_list=${14}
 
-            echo "Running TEP mode ($mtp_mode) with ctx_num=$ctx_num, gen_num=$gen_num, gen_tp_size=$gen_tp_size, gen_ep_size=$gen_ep_size, gen_batch_size=$gen_batch_size, gen_max_num_tokens=$gen_max_num_tokens, gen_gpu_memory_fraction=$gen_gpu_memory_fraction, gen_mtp_size=$gen_mtp_size, gen_eplb_num_slots=$gen_eplb_num_slots, gen_concurrency_list=\"$gen_concurrency_list\""
+            echo "Running TEP mode ($mtp_mode) with ctx_num=$ctx_num, gen_num=$gen_num, gen_tp_size=$gen_tp_size, gen_ep_size=$gen_tp_size, gen_batch_size=$gen_batch_size, gen_max_num_tokens=$gen_max_num_tokens, gen_gpu_memory_fraction=$gen_gpu_memory_fraction, gen_mtp_size=$gen_mtp_size, gen_eplb_num_slots=$gen_eplb_num_slots, gen_concurrency_list=\"$gen_concurrency_list\""
 
             # TEP mode: Use false to disable attention dp
-            run_single $ctx_num $ctx_tp_size $ctx_ep_size $ctx_enable_attention_dp $gen_num $gen_tp_size $gen_ep_size $gen_batch_size $gen_max_num_tokens false $gen_gpu_memory_fraction $gen_mtp_size $gen_eplb_num_slots "$gen_concurrency_list"
+            run_single $ctx_num $ctx_tp_size $ctx_ep_size $ctx_enable_attention_dp $gen_num $gen_tp_size $gen_tp_size $gen_batch_size $gen_max_num_tokens false $gen_gpu_memory_fraction $gen_mtp_size $gen_eplb_num_slots "$gen_concurrency_list"
             ;;
         "dep")
-            if [ $# -ne 15 ]; then
-                echo "Error: DEP mode requires 15 additional parameters (including mtp_mode)"
+            if [ $# -ne 14 ]; then
+                echo "Error: DEP mode requires 14 additional parameters (including mtp_mode)"
                 usage
             fi
 
@@ -409,17 +408,16 @@ main() {
             local ctx_enable_attention_dp=$6
             local gen_num=$7
             local gen_tp_size=$8
-            local gen_ep_size=$9
-            local gen_batch_size=${10}
-            local gen_max_num_tokens=${11}
-            local gen_gpu_memory_fraction=${12}
-            local gen_mtp_size=${13}
-            local gen_eplb_num_slots=${14}
-            local gen_concurrency_list=${15}
+            local gen_batch_size=$9
+            local gen_max_num_tokens=${10}
+            local gen_gpu_memory_fraction=${11}
+            local gen_mtp_size=${12}
+            local gen_eplb_num_slots=${13}
+            local gen_concurrency_list=${14}
 
-            echo "Running DEP mode ($mtp_mode) with ctx_num=$ctx_num, ctx_tp_size=$ctx_tp_size, ctx_enable_attention_dp=$ctx_enable_attention_dp, gen_num=$gen_num, gen_tp_size=$gen_tp_size, gen_ep_size=$gen_ep_size, gen_batch_size=$gen_batch_size, gen_max_num_tokens=$gen_max_num_tokens, gen_gpu_memory_fraction=$gen_gpu_memory_fraction, gen_mtp_size=$gen_mtp_size, gen_eplb_num_slots=$gen_eplb_num_slots, gen_concurrency_list=\"$gen_concurrency_list\""
+            echo "Running DEP mode ($mtp_mode) with ctx_num=$ctx_num, ctx_tp_size=$ctx_tp_size, ctx_enable_attention_dp=$ctx_enable_attention_dp, gen_num=$gen_num, gen_tp_size=$gen_tp_size, gen_ep_size=$gen_tp_size, gen_batch_size=$gen_batch_size, gen_max_num_tokens=$gen_max_num_tokens, gen_gpu_memory_fraction=$gen_gpu_memory_fraction, gen_mtp_size=$gen_mtp_size, gen_eplb_num_slots=$gen_eplb_num_slots, gen_concurrency_list=\"$gen_concurrency_list\""
 
-            run_single $ctx_num $ctx_tp_size $ctx_ep_size $ctx_enable_attention_dp $gen_num $gen_tp_size $gen_ep_size $gen_batch_size $gen_max_num_tokens $gen_gpu_memory_fraction $gen_mtp_size $gen_eplb_num_slots "$gen_concurrency_list"
+            run_single $ctx_num $ctx_tp_size $ctx_ep_size $ctx_enable_attention_dp $gen_num $gen_tp_size $gen_tp_size $gen_batch_size $gen_max_num_tokens $gen_gpu_memory_fraction $gen_mtp_size $gen_eplb_num_slots "$gen_concurrency_list"
             ;;
         *)
             echo "Error: Unknown mode '$mode'"