|
35 | 35 | from lightllm.server.router.model_infer.mode_backend.mtp_pre_process import ( |
36 | 36 | prepare_mtp_prefill_inputs, |
37 | 37 | ) |
38 | | -from lightllm.utils.config_utils import get_dtype, get_vocab_size |
| 38 | +from lightllm.utils.config_utils import auto_set_fused_shared_experts, get_dtype, get_vocab_size |
39 | 39 | from lightllm.utils.dist_utils import init_distributed_env |
40 | 40 | from lightllm.utils.envs_utils import set_env_start_args |
41 | 41 |
|
@@ -551,6 +551,7 @@ def _make_decode_input( |
551 | 551 | b_req_idx=req_idx, |
552 | 552 | b_mtp_index=mtp_index, |
553 | 553 | b_seq_len=seq_len, |
| 554 | + b_position_delta=cpu_i32_zeros(batch_size), |
554 | 555 | mem_indexes_cpu=mem_indexes, |
555 | 556 | is_prefill=False, |
556 | 557 | multimodal_params=empty_multimodal_params(batch_size), |
@@ -638,6 +639,7 @@ def _slice_decode_input(self, model_input: ModelInput, batch_start: int, batch_e |
638 | 639 | b_req_idx=model_input.b_req_idx[batch_start:batch_end].clone(), |
639 | 640 | b_mtp_index=model_input.b_mtp_index[batch_start:batch_end].clone(), |
640 | 641 | b_seq_len=b_seq_len, |
| 642 | + b_position_delta=model_input.b_position_delta[batch_start:batch_end].clone(), |
641 | 643 | mem_indexes_cpu=model_input.mem_indexes_cpu[batch_start:batch_end].contiguous(), |
642 | 644 | is_prefill=False, |
643 | 645 | multimodal_params=model_input.multimodal_params[batch_start:batch_end], |
@@ -1409,6 +1411,7 @@ def main(argv: Optional[Sequence[str]] = None): |
1409 | 1411 | parser = make_argument_parser() |
1410 | 1412 | add_static_benchmark_args(parser) |
1411 | 1413 | args = parser.parse_args(argv) |
| 1414 | + auto_set_fused_shared_experts(args) |
1412 | 1415 | if args.benchmark in {"all", "prefill"} and args.batch_max_tokens is None: |
1413 | 1416 | args.batch_max_tokens = 8192 |
1414 | 1417 | cases = build_cases(args) |
|
0 commit comments