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=}"
|
), 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
|
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_kwargs = dict(
|
||||||
gsp_num_groups=bench_args.gsp_num_groups,
|
gsp_num_groups=bench_args.gsp_num_groups,
|
||||||
gsp_system_prompt_len=bench_args.gsp_system_prompt_len,
|
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,
|
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
|
# Warmup
|
||||||
if not bench_args.skip_warmup:
|
if not bench_args.skip_warmup:
|
||||||
batch_size_unique = list(set(bench_args.batch_size))
|
batch_size_unique = list(set(bench_args.batch_size))
|
||||||
|
|||||||
Reference in New Issue
Block a user