fix: revert change to aiperf output format (#3837)
Signed-off-by: hongkuanz <hongkuanz@nvidia.com>
This commit is contained in:
parent
c6b5904579
commit
fa45ff1f8a
|
|
@ -208,16 +208,11 @@ class LoadGenerator:
|
|||
logger.info(f"Parsing results from: {results_file}")
|
||||
|
||||
with open(results_file, "r") as f:
|
||||
data = json.load(f)
|
||||
|
||||
# aiperf now wraps metrics under "records" key
|
||||
metrics = data.get("records", data)
|
||||
metrics = json.load(f)
|
||||
|
||||
results = {
|
||||
"throughput": metrics.get("output_token_throughput", {}).get("avg", 0),
|
||||
"ttft_mean": metrics.get("ttft", {}).get(
|
||||
"avg", 0
|
||||
), # Changed from "time_to_first_token"
|
||||
"ttft_mean": metrics.get("time_to_first_token", {}).get("avg", 0),
|
||||
"itl_mean": metrics.get("inter_token_latency", {}).get("avg", 0),
|
||||
"end_to_end_latency_mean": metrics.get("request_latency", {}).get(
|
||||
"avg", 0
|
||||
|
|
|
|||
Loading…
Reference in New Issue