Fix bench_one_batch_server by moving the print statements (#19175)
This commit is contained in:
@@ -803,6 +803,11 @@ def run_benchmark_internal(
|
||||
), f"effective_max_running_requests_per_dp is not set, {max_running_requests_per_dp=}"
|
||||
skip_max_running_requests_threshold = max_running_requests_per_dp * dp_size
|
||||
|
||||
print(f"{max_running_requests_per_dp=}")
|
||||
print(f"{dp_size=}")
|
||||
print(f"{skip_max_running_requests_threshold=}")
|
||||
print(f"{skip_token_capacity_threshold=}")
|
||||
|
||||
gsp_kwargs = dict(
|
||||
gsp_num_groups=bench_args.gsp_num_groups,
|
||||
gsp_system_prompt_len=bench_args.gsp_system_prompt_len,
|
||||
@@ -810,11 +815,6 @@ def run_benchmark_internal(
|
||||
gsp_output_len=bench_args.gsp_output_len,
|
||||
)
|
||||
|
||||
print(f"{max_running_requests_per_dp=}")
|
||||
print(f"{dp_size=}")
|
||||
print(f"{skip_max_running_requests_threshold=}")
|
||||
print(f"{skip_token_capacity_threshold=}")
|
||||
|
||||
# Warmup
|
||||
if not bench_args.skip_warmup:
|
||||
batch_size_unique = list(set(bench_args.batch_size))
|
||||
|
||||
Reference in New Issue
Block a user