From 46f0f003b8ccfd42ff0833eacc95d9bd8c089eb4 Mon Sep 17 00:00:00 2001 From: Yang Wang Date: Mon, 10 Nov 2025 13:56:33 -0800 Subject: [PATCH] add config --- .../benchmark_v3/configs/configurations.tsx | 12 +++++++++++- torchci/pages/api/benchmark/get_time_series.ts | 1 + 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/torchci/components/benchmark_v3/configs/configurations.tsx b/torchci/components/benchmark_v3/configs/configurations.tsx index a0e7314282..0bb270d281 100644 --- a/torchci/components/benchmark_v3/configs/configurations.tsx +++ b/torchci/components/benchmark_v3/configs/configurations.tsx @@ -26,8 +26,12 @@ export const BENCHMARK_ID_MAPPING: Record = { repoName: "pytorch/helion", benchmarkName: "Helion Benchmark", }, + torchao_micro_api_benchmark: { + id: "torchao_micro_api_benchmark", + repoName: "pytorch/ao", + benchmarkName: "micro-benchmark api", + }, }; - /** * A helper function to get benchmark id from report id * @param reportId @@ -119,6 +123,12 @@ export const BENCHMARK_CATEGORIES: BenchmarkCategoryGroup[] = [ route: "/benchmark/llms?repoName=pytorch%2Fao&benchmarkName=micro-benchmark+api", info: "Powered by [code](https://github.com/pytorch/ao/blob/main/docs/source/benchmarking_api_guide.md)", + actions: [ + { + label: "New dashboard (WIP)", + href: "/benchmark/v3/dashboard/torchao_micro_api_benchmark", + }, + ], }, ], }, diff --git a/torchci/pages/api/benchmark/get_time_series.ts b/torchci/pages/api/benchmark/get_time_series.ts index eb419f3904..3339d04d8d 100644 --- a/torchci/pages/api/benchmark/get_time_series.ts +++ b/torchci/pages/api/benchmark/get_time_series.ts @@ -79,6 +79,7 @@ async function getBenchmarkTimeSeriesData( ); case "pytorch_operator_microbenchmark": case "pytorch_helion": + case "torchao_micro_api_benchmark": return await getGenernalBenchmarkTimeSeries( query_params, formats,