Skip to content

Commit 9306b81

Browse files
authored
Create Helion Benchmark (#7097)
### Create Helion Benchmark
1 parent cc6da23 commit 9306b81

File tree

2 files changed

+19
-0
lines changed

2 files changed

+19
-0
lines changed

torchci/components/layout/NavBar.tsx

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,10 @@ function NavBar() {
122122
name: "SGLang",
123123
href: "/benchmark/llms?repoName=sgl-project%2Fsglang",
124124
},
125+
{
126+
name: "Helion",
127+
href: "/benchmark/llms?repoName=pytorch%2Fhelion&benchmarkName=Helion+Benchmark",
128+
},
125129
];
126130

127131
const devInfraDropdown = [

torchci/lib/benchmark/llms/common.ts

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,12 @@ export const METRIC_DISPLAY_HEADERS: { [k: string]: string } = {
4242
p99_ttft_ms: "p99 TTFT (ms)",
4343
requests_per_second: "Requests/s",
4444
tokens_per_second: "Tokens/s",
45+
triton_speedup: "Triton Speedup",
46+
triton_accuracy: "Triton Accuracy",
47+
torch_compile_speedup: "Torch Compile Speedup",
48+
torch_compile_accuracy: "Torch Compile Accuracy",
49+
helion_speedup: "Helion Speedup",
50+
helion_accuracy: "Helion Accuracy",
4551
};
4652
// The variable name is a bit dumb, but it tells if a higher metric value
4753
// is good or bad so that we can highlight it on the dashboard accordingly.
@@ -71,6 +77,12 @@ export const IS_INCREASING_METRIC_VALUE_GOOD: { [k: string]: boolean } = {
7177
"Warm compile time (s)": false,
7278
Speedup: true,
7379
"Speedup (%)": true,
80+
triton_speedup: true,
81+
triton_accuracy: true,
82+
torch_compile_speedup: true,
83+
torch_compile_accuracy: true,
84+
helion_speedup: true,
85+
helion_accuracy: true,
7486
};
7587
export const METRIC_DISPLAY_SHORT_HEADERS: { [k: string]: string } = {
7688
"memory_bandwidth(GB/s)": "Bandwidth",
@@ -87,6 +99,9 @@ export const METRIC_DISPLAY_SHORT_HEADERS: { [k: string]: string } = {
8799
};
88100
export const UNIT_FOR_METRIC: { [k: string]: string } = {
89101
"Speedup (%)": "%",
102+
triton_speedup: "x",
103+
torch_compile_speedup: "x",
104+
helion_speedup: "x",
90105
};
91106

92107
export const DEFAULT_DEVICE_NAME = "All Devices";

0 commit comments

Comments
 (0)