Spaces:
Sleeping
Sleeping
Commit
·
48bb3eb
0
Parent(s):
Initial commit
Browse files- .gitignore +13 -0
- .python-version +1 -0
- CLAUDE.md +49 -0
- README.md +45 -0
- app.py +150 -0
- populate_dataset.py +69 -0
- pyproject.toml +13 -0
- src/utils.py +64 -0
- uv.lock +0 -0
.gitignore
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Python-generated files
|
| 2 |
+
__pycache__/
|
| 3 |
+
*.py[oc]
|
| 4 |
+
build/
|
| 5 |
+
dist/
|
| 6 |
+
wheels/
|
| 7 |
+
*.egg-info
|
| 8 |
+
|
| 9 |
+
# Virtual environments
|
| 10 |
+
.venv
|
| 11 |
+
|
| 12 |
+
.gradio
|
| 13 |
+
|
.python-version
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
3.13
|
CLAUDE.md
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# CLAUDE.md
|
| 2 |
+
|
| 3 |
+
This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
|
| 4 |
+
|
| 5 |
+
## Project Overview
|
| 6 |
+
|
| 7 |
+
This is a Python project called "amd-leaderboard" that appears to be designed for tracking performance metrics or benchmarks in a leaderboard format.
|
| 8 |
+
|
| 9 |
+
## Development Environment
|
| 10 |
+
|
| 11 |
+
- **Python Version**: 3.13 (specified in `.python-version`)
|
| 12 |
+
- **Package Manager**: uv (modern Python package manager)
|
| 13 |
+
- **Project Configuration**: `pyproject.toml`
|
| 14 |
+
|
| 15 |
+
## Common Commands
|
| 16 |
+
|
| 17 |
+
### Package Management
|
| 18 |
+
```bash
|
| 19 |
+
# Install dependencies
|
| 20 |
+
uv sync
|
| 21 |
+
|
| 22 |
+
# Add a new dependency
|
| 23 |
+
uv add <package_name>
|
| 24 |
+
|
| 25 |
+
# Add a development dependency
|
| 26 |
+
uv add --dev <package_name>
|
| 27 |
+
```
|
| 28 |
+
|
| 29 |
+
### Running the Application
|
| 30 |
+
```bash
|
| 31 |
+
# Run the main application
|
| 32 |
+
uv run python main.py
|
| 33 |
+
|
| 34 |
+
# Run any Python file with the project's environment
|
| 35 |
+
uv run python <file_path>
|
| 36 |
+
```
|
| 37 |
+
|
| 38 |
+
## Project Structure
|
| 39 |
+
|
| 40 |
+
The codebase follows a simple Python package structure:
|
| 41 |
+
- `main.py` - Main entry point (currently empty)
|
| 42 |
+
- `src/` - Source code directory
|
| 43 |
+
- `utils.py` - Contains a results data structure template
|
| 44 |
+
|
| 45 |
+
## Key Information
|
| 46 |
+
|
| 47 |
+
1. The project uses modern Python tooling with `uv` as the package manager and `pyproject.toml` for configuration
|
| 48 |
+
2. Currently has no external dependencies installed
|
| 49 |
+
3. The `src/utils.py` file contains a template structure that suggests the project will handle results with task names and metric values in a dictionary format
|
README.md
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
title: AMD vLLM Benchmark Leaderboard
|
| 3 |
+
emoji: <�
|
| 4 |
+
colorFrom: red
|
| 5 |
+
colorTo: blue
|
| 6 |
+
sdk: gradio
|
| 7 |
+
sdk_version: 4.44.1
|
| 8 |
+
app_file: app.py
|
| 9 |
+
pinned: false
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# AMD vLLM Benchmark Leaderboard
|
| 13 |
+
|
| 14 |
+
A leaderboard for tracking performance and accuracy metrics of vLLM benchmarks on AMD hardware.
|
| 15 |
+
|
| 16 |
+
## Features
|
| 17 |
+
|
| 18 |
+
- **Performance Metrics Tracking**: TTFT, TPOT, ITL, E2E Latency, and Throughput
|
| 19 |
+
- **Accuracy Metrics**: WikiText perplexity scores
|
| 20 |
+
- **Manual Result Submission**: Enter benchmark results through the web interface
|
| 21 |
+
- **JSON Upload**: Direct upload of benchmark result files from vLLM
|
| 22 |
+
- **Sortable Leaderboard**: Compare results across different runs and configurations
|
| 23 |
+
|
| 24 |
+
## Metrics
|
| 25 |
+
|
| 26 |
+
- **TTFT (Time To First Token)**: Time to generate the first token (ms)
|
| 27 |
+
- **TPOT (Time Per Output Token)**: Average time per token generation (ms)
|
| 28 |
+
- **ITL (Inter-Token Latency)**: Latency between tokens (ms)
|
| 29 |
+
- **E2E Latency**: End-to-end request latency (ms)
|
| 30 |
+
- **Throughput**: Total tokens processed per second
|
| 31 |
+
- **WikiText Perplexity**: Model accuracy metric (lower is better)
|
| 32 |
+
|
| 33 |
+
## Usage
|
| 34 |
+
|
| 35 |
+
1. **View Results**: Check the Leaderboard tab to see all benchmark results
|
| 36 |
+
2. **Submit Results**: Use the Submit Results tab to add new benchmark data
|
| 37 |
+
- Manual entry for individual metrics
|
| 38 |
+
- JSON upload for direct benchmark output files
|
| 39 |
+
3. **Compare**: Sort and analyze results to identify best configurations
|
| 40 |
+
|
| 41 |
+
## Integration with vLLM Benchmarks
|
| 42 |
+
|
| 43 |
+
This leaderboard is designed to work with output from:
|
| 44 |
+
- `vllm/benchmarks/benchmark_serving.py` for performance metrics
|
| 45 |
+
- `lm-evaluation-harness` for accuracy metrics
|
app.py
ADDED
|
@@ -0,0 +1,150 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
import pandas as pd
|
| 3 |
+
from datetime import datetime
|
| 4 |
+
import os
|
| 5 |
+
from datasets import load_dataset, Dataset
|
| 6 |
+
|
| 7 |
+
# Configuration
|
| 8 |
+
DATASET_ID = "siro1/amd-hackathon"
|
| 9 |
+
HF_TOKEN = os.environ.get("HF_TOKEN") # Optional: for write access
|
| 10 |
+
|
| 11 |
+
dataset = None
|
| 12 |
+
|
| 13 |
+
|
| 14 |
+
def load_results():
|
| 15 |
+
global dataset
|
| 16 |
+
"""Load results from Hugging Face dataset"""
|
| 17 |
+
dataset = load_dataset(DATASET_ID, split="train")
|
| 18 |
+
# Convert dataset to dictionary format matching the expected structure
|
| 19 |
+
results = dataset.map(
|
| 20 |
+
lambda item: {
|
| 21 |
+
"Team": item["team"],
|
| 22 |
+
"Timestamp": item["timestamp"],
|
| 23 |
+
"TTFT (ms)": item["ttft"],
|
| 24 |
+
"TPOT (ms)": item["tpot"],
|
| 25 |
+
"ITL (ms)": item["itl"],
|
| 26 |
+
"E2E Latency (ms)": item["e2e"],
|
| 27 |
+
"Throughput (tokens/s)": item["throughput"],
|
| 28 |
+
"Bits per Byte": item["bits_per_byte"],
|
| 29 |
+
"Byte Perplexity": item["byte_perplexity"],
|
| 30 |
+
"Word Perplexity": item["word_perplexity"],
|
| 31 |
+
},
|
| 32 |
+
batch_size=64,
|
| 33 |
+
remove_columns=dataset.column_names,
|
| 34 |
+
)
|
| 35 |
+
df = results.to_pandas()
|
| 36 |
+
df = df.sort_values("Throughput (tokens/s)", ascending=False)
|
| 37 |
+
return df
|
| 38 |
+
|
| 39 |
+
|
| 40 |
+
def update_dataset(
|
| 41 |
+
team_name,
|
| 42 |
+
ttft,
|
| 43 |
+
tpot,
|
| 44 |
+
itl,
|
| 45 |
+
e2e,
|
| 46 |
+
throughput,
|
| 47 |
+
bits_per_byte,
|
| 48 |
+
byte_perplexity,
|
| 49 |
+
word_perplexity,
|
| 50 |
+
):
|
| 51 |
+
"""Insert a new row into the Hugging Face dataset"""
|
| 52 |
+
existing_data = dataset.to_list()
|
| 53 |
+
|
| 54 |
+
new_entry = {
|
| 55 |
+
"team": team_name,
|
| 56 |
+
"timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
| 57 |
+
"ttft": float(ttft),
|
| 58 |
+
"tpot": float(tpot),
|
| 59 |
+
"itl": float(itl),
|
| 60 |
+
"e2e": float(e2e),
|
| 61 |
+
"throughput": float(throughput),
|
| 62 |
+
"bits_per_byte": float(bits_per_byte),
|
| 63 |
+
"byte_perplexity": float(byte_perplexity),
|
| 64 |
+
"word_perplexity": float(word_perplexity),
|
| 65 |
+
}
|
| 66 |
+
existing_data.append(new_entry)
|
| 67 |
+
updated_dataset = Dataset.from_list(existing_data)
|
| 68 |
+
updated_dataset.push_to_hub(DATASET_ID, token=HF_TOKEN)
|
| 69 |
+
|
| 70 |
+
return True
|
| 71 |
+
|
| 72 |
+
|
| 73 |
+
def api_submit_results(
|
| 74 |
+
team_name: str,
|
| 75 |
+
ttft: float,
|
| 76 |
+
tpot: float,
|
| 77 |
+
itl: float,
|
| 78 |
+
e2e: float,
|
| 79 |
+
throughput: float,
|
| 80 |
+
bits_per_byte: float,
|
| 81 |
+
byte_perplexity: float,
|
| 82 |
+
word_perplexity: float,
|
| 83 |
+
) -> str:
|
| 84 |
+
try:
|
| 85 |
+
# Update the dataset with new submission
|
| 86 |
+
success = update_dataset(
|
| 87 |
+
team_name=team_name,
|
| 88 |
+
ttft=ttft,
|
| 89 |
+
tpot=tpot,
|
| 90 |
+
itl=itl,
|
| 91 |
+
e2e=e2e,
|
| 92 |
+
throughput=throughput,
|
| 93 |
+
bits_per_byte=bits_per_byte,
|
| 94 |
+
byte_perplexity=byte_perplexity,
|
| 95 |
+
word_perplexity=word_perplexity,
|
| 96 |
+
)
|
| 97 |
+
|
| 98 |
+
if success:
|
| 99 |
+
return f"Your submission for {team_name} has been accepted 🤗"
|
| 100 |
+
else:
|
| 101 |
+
return f"Failed to submit results for {team_name} 😢"
|
| 102 |
+
|
| 103 |
+
except Exception as e:
|
| 104 |
+
return f"Failed to submit results for {team_name} 😢: {str(e)}"
|
| 105 |
+
|
| 106 |
+
|
| 107 |
+
# Create Gradio interface
|
| 108 |
+
def create_interface():
|
| 109 |
+
with gr.Blocks(title="AMD vLLM Benchmark Leaderboard") as demo:
|
| 110 |
+
gr.Markdown("# AMD vLLM Benchmark Leaderboard")
|
| 111 |
+
gr.Markdown(
|
| 112 |
+
"Track and compare performance and accuracy metrics for vLLM benchmarks"
|
| 113 |
+
)
|
| 114 |
+
|
| 115 |
+
with gr.Tab("Leaderboard"):
|
| 116 |
+
# Function to refresh leaderboard
|
| 117 |
+
def refresh_leaderboard():
|
| 118 |
+
return load_results()
|
| 119 |
+
|
| 120 |
+
# Initial load
|
| 121 |
+
leaderboard_table = gr.DataFrame(
|
| 122 |
+
value=refresh_leaderboard(),
|
| 123 |
+
label="Benchmark Results",
|
| 124 |
+
interactive=False,
|
| 125 |
+
)
|
| 126 |
+
|
| 127 |
+
refresh_btn = gr.Button("Refresh Leaderboard")
|
| 128 |
+
refresh_btn.click(
|
| 129 |
+
fn=refresh_leaderboard,
|
| 130 |
+
outputs=leaderboard_table,
|
| 131 |
+
)
|
| 132 |
+
|
| 133 |
+
gr.api(
|
| 134 |
+
fn=api_submit_results,
|
| 135 |
+
api_name="submit_results",
|
| 136 |
+
)
|
| 137 |
+
|
| 138 |
+
return demo
|
| 139 |
+
|
| 140 |
+
|
| 141 |
+
# Create and launch the app
|
| 142 |
+
if __name__ == "__main__":
|
| 143 |
+
demo = create_interface()
|
| 144 |
+
|
| 145 |
+
demo.queue()
|
| 146 |
+
demo.launch(
|
| 147 |
+
server_name="0.0.0.0",
|
| 148 |
+
server_port=7860,
|
| 149 |
+
share=True,
|
| 150 |
+
)
|
populate_dataset.py
ADDED
|
@@ -0,0 +1,69 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env python3
|
| 2 |
+
"""
|
| 3 |
+
Script to populate the Hugging Face dataset with mock data
|
| 4 |
+
"""
|
| 5 |
+
|
| 6 |
+
from datasets import Dataset
|
| 7 |
+
from datetime import datetime, timedelta
|
| 8 |
+
import random
|
| 9 |
+
|
| 10 |
+
# Configuration
|
| 11 |
+
DATASET_ID = "siro1/amd-hackathon"
|
| 12 |
+
HF_TOKEN = None # Set this if needed for private repos
|
| 13 |
+
|
| 14 |
+
# Generate mock data
|
| 15 |
+
mock_data = []
|
| 16 |
+
|
| 17 |
+
teams = ["Team Alpha", "Team Beta", "Team Gamma", "Team Delta", "Team Epsilon"]
|
| 18 |
+
base_date = datetime.now() - timedelta(days=7)
|
| 19 |
+
|
| 20 |
+
for i in range(10):
|
| 21 |
+
team = random.choice(teams)
|
| 22 |
+
timestamp = (base_date + timedelta(days=i / 2)).strftime("%Y-%m-%d %H:%M:%S")
|
| 23 |
+
|
| 24 |
+
# Vary configurations
|
| 25 |
+
input_length = random.choice([128, 256, 512])
|
| 26 |
+
output_length = random.choice([128, 256, 512])
|
| 27 |
+
concurrent_requests = random.choice([8, 16, 32, 64])
|
| 28 |
+
|
| 29 |
+
# Generate performance metrics with some variance
|
| 30 |
+
base_ttft = 40 + random.uniform(-10, 10)
|
| 31 |
+
base_tpot = 11 + random.uniform(-2, 2)
|
| 32 |
+
base_itl = 10 + random.uniform(-2, 2)
|
| 33 |
+
base_e2e = 1500 + (input_length + output_length) * 2 + random.uniform(-200, 200)
|
| 34 |
+
base_throughput = 2000 + concurrent_requests * 20 + random.uniform(-200, 200)
|
| 35 |
+
bits_per_byte = 0.54 + random.uniform(-0.02, 0.02)
|
| 36 |
+
byte_perplexity = 1.45 + random.uniform(-0.02, 0.02)
|
| 37 |
+
word_perplexity = 4.13 + random.uniform(-0.02, 0.02)
|
| 38 |
+
|
| 39 |
+
entry = {
|
| 40 |
+
"team": team,
|
| 41 |
+
"timestamp": timestamp,
|
| 42 |
+
"ttft": round(base_ttft, 2),
|
| 43 |
+
"tpot": round(base_tpot, 2),
|
| 44 |
+
"itl": round(base_itl, 2),
|
| 45 |
+
"e2e": round(base_e2e, 2),
|
| 46 |
+
"throughput": round(base_throughput, 2),
|
| 47 |
+
"bits_per_byte": round(bits_per_byte, 2),
|
| 48 |
+
"byte_perplexity": round(byte_perplexity, 2),
|
| 49 |
+
"word_perplexity": round(word_perplexity, 2),
|
| 50 |
+
}
|
| 51 |
+
|
| 52 |
+
mock_data.append(entry)
|
| 53 |
+
|
| 54 |
+
# Sort by timestamp
|
| 55 |
+
mock_data.sort(key=lambda x: x["timestamp"])
|
| 56 |
+
|
| 57 |
+
# Create dataset and push to hub
|
| 58 |
+
print(f"Creating dataset with {len(mock_data)} entries...")
|
| 59 |
+
dataset = Dataset.from_list(mock_data)
|
| 60 |
+
|
| 61 |
+
print(f"Pushing to Hugging Face Hub: {DATASET_ID}")
|
| 62 |
+
dataset.push_to_hub(DATASET_ID, token=HF_TOKEN)
|
| 63 |
+
|
| 64 |
+
print("Dataset populated successfully!")
|
| 65 |
+
print("\nSample entries:")
|
| 66 |
+
for entry in mock_data[:3]:
|
| 67 |
+
print(
|
| 68 |
+
f"- {entry['team']} at {entry['timestamp']}: throughput={entry['throughput']}"
|
| 69 |
+
)
|
pyproject.toml
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[project]
|
| 2 |
+
name = "amd-leaderboard"
|
| 3 |
+
version = "0.1.0"
|
| 4 |
+
description = "A leaderboard for tracking vLLM benchmark results on AMD hardware"
|
| 5 |
+
readme = "README.md"
|
| 6 |
+
requires-python = ">=3.13"
|
| 7 |
+
dependencies = [
|
| 8 |
+
"datasets>=3.6.0",
|
| 9 |
+
"gradio>=5.35.0",
|
| 10 |
+
"huggingface-hub>=0.33.1",
|
| 11 |
+
"ipython>=9.3.0",
|
| 12 |
+
"pandas>=2.3.0",
|
| 13 |
+
]
|
src/utils.py
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Example result structure for vLLM benchmarks
|
| 2 |
+
RESULT_SCHEMA = {
|
| 3 |
+
"run_id": {
|
| 4 |
+
"model": "model_name",
|
| 5 |
+
"timestamp": "YYYY-MM-DD HH:MM:SS",
|
| 6 |
+
"config": {
|
| 7 |
+
"input_length": 128,
|
| 8 |
+
"output_length": 128,
|
| 9 |
+
"concurrent_requests": 16
|
| 10 |
+
},
|
| 11 |
+
"performance": {
|
| 12 |
+
"median_ttft_ms": 0.0, # Time To First Token
|
| 13 |
+
"median_tpot_ms": 0.0, # Time Per Output Token
|
| 14 |
+
"median_itl_ms": 0.0, # Inter-Token Latency
|
| 15 |
+
"median_e2el_ms": 0.0, # End-to-End Latency
|
| 16 |
+
"total_token_throughput": 0.0
|
| 17 |
+
},
|
| 18 |
+
"accuracy": {
|
| 19 |
+
"wikitext_perplexity": 0.0
|
| 20 |
+
}
|
| 21 |
+
}
|
| 22 |
+
}
|
| 23 |
+
|
| 24 |
+
# Mock data for testing
|
| 25 |
+
MOCK_RESULTS = {
|
| 26 |
+
"run_2024_12_01_baseline": {
|
| 27 |
+
"model": "amd/Mixtral-8x7B-Instruct-v0.1-FP8-KV",
|
| 28 |
+
"timestamp": "2024-12-01 10:30:00",
|
| 29 |
+
"config": {
|
| 30 |
+
"input_length": 128,
|
| 31 |
+
"output_length": 128,
|
| 32 |
+
"concurrent_requests": 16
|
| 33 |
+
},
|
| 34 |
+
"performance": {
|
| 35 |
+
"median_ttft_ms": 45.23,
|
| 36 |
+
"median_tpot_ms": 12.56,
|
| 37 |
+
"median_itl_ms": 11.89,
|
| 38 |
+
"median_e2el_ms": 1589.45,
|
| 39 |
+
"total_token_throughput": 2048.67
|
| 40 |
+
},
|
| 41 |
+
"accuracy": {
|
| 42 |
+
"wikitext_perplexity": 7.89
|
| 43 |
+
}
|
| 44 |
+
},
|
| 45 |
+
"run_2024_12_02_optimized": {
|
| 46 |
+
"model": "amd/Mixtral-8x7B-Instruct-v0.1-FP8-KV",
|
| 47 |
+
"timestamp": "2024-12-02 14:15:00",
|
| 48 |
+
"config": {
|
| 49 |
+
"input_length": 128,
|
| 50 |
+
"output_length": 128,
|
| 51 |
+
"concurrent_requests": 16
|
| 52 |
+
},
|
| 53 |
+
"performance": {
|
| 54 |
+
"median_ttft_ms": 42.11,
|
| 55 |
+
"median_tpot_ms": 11.23,
|
| 56 |
+
"median_itl_ms": 10.95,
|
| 57 |
+
"median_e2el_ms": 1456.78,
|
| 58 |
+
"total_token_throughput": 2234.89
|
| 59 |
+
},
|
| 60 |
+
"accuracy": {
|
| 61 |
+
"wikitext_perplexity": 7.91
|
| 62 |
+
}
|
| 63 |
+
}
|
| 64 |
+
}
|
uv.lock
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|