[torchao] Fix torchao GraphPanel (#5302)

- Remove the incorrect HTTP request to compiler_benchmark_performance - Fix the GraphPanel on the summary page.
pytorch · Jun 6, 2024 · b4172e5 · b4172e5
1 parent 270b951
commit b4172e5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/torchci/components/benchmark/compilers/SummaryGraphPanel.tsx b/torchci/components/benchmark/compilers/SummaryGraphPanel.tsx
@@ -23,13 +23,15 @@ import useSWR from "swr";
 const GRAPH_ROW_HEIGHT = 245;
 
 export function GraphPanel({
+  queryName,
   queryParams,
   granularity,
   suite,
   branch,
   lCommit,
   rCommit,
 }: {
+  queryName: string;
   queryParams: RocksetParam[];
   granularity: Granularity;
   suite: string;
@@ -41,6 +43,7 @@ export function GraphPanel({
   // from Rockset small enough (<6MB) to fit into Vercel lambda limit
   return (
     <SuiteGraphPanel
+      queryName={queryName}
       queryParams={queryParams}
       granularity={granularity}
       suite={suite}
@@ -52,13 +55,15 @@ export function GraphPanel({
 }
 
 function SuiteGraphPanel({
+  queryName,
   queryParams,
   granularity,
   suite,
   branch,
   lCommit,
   rCommit,
 }: {
+  queryName: string;
   queryParams: RocksetParam[];
   granularity: Granularity;
   suite: string;
@@ -67,7 +72,6 @@ function SuiteGraphPanel({
   rCommit: string;
 }) {
   const queryCollection = "inductor";
-  const queryName = "compilers_benchmark_performance";
 
   const queryParamsWithSuite: RocksetParam[] = [
     {

diff --git a/torchci/pages/benchmark/compilers.tsx b/torchci/pages/benchmark/compilers.tsx
@@ -151,6 +151,7 @@ function Report({
         all_suites={SUITES}
       />
       <GraphPanel
+        queryName={"compilers_benchmark_performance"}
         queryParams={queryParams}
         granularity={granularity}
         suite={suite}

diff --git a/torchci/pages/benchmark/torchao.tsx b/torchci/pages/benchmark/torchao.tsx
@@ -147,6 +147,7 @@ function Report({
         all_suites={SUITES}
       />
       <GraphPanel
+        queryName={"torchao_query"}
         queryParams={queryParams}
         granularity={granularity}
         suite={suite}