use exact date matching for GPU comparison benchmark queries

Author: adibarra

packages/app/src/app/api/v1/benchmarks/route.ts

@@ -9,9 +9,9 @@ import { cachedJson, cachedQuery } from '@/lib/api-cache';
 export const dynamic = 'force-dynamic';
 
 const getCachedBenchmarks = cachedQuery(
-  async (dbModelKey: string, date?: string) => {
+  async (dbModelKey: string, date?: string, exact?: boolean) => {
     const sql = getDb();
-    return getLatestBenchmarks(sql, dbModelKey, date);
+    return getLatestBenchmarks(sql, dbModelKey, date, exact);
   },
   'benchmarks',
   { blobOnly: true },
@@ -21,13 +21,14 @@ export async function GET(request: NextRequest) {
   const params = request.nextUrl.searchParams;
   const model = params.get('model') ?? '';
   const date = params.get('date') ?? undefined;
+  const exact = params.get('exact') === 'true';
   const dbModelKey = DISPLAY_MODEL_TO_DB[model];
   if (!dbModelKey) {
     return NextResponse.json({ error: 'Unknown model' }, { status: 400 });
   }
 
   try {
-    const rows = await getCachedBenchmarks(dbModelKey, date);
+    const rows = await getCachedBenchmarks(dbModelKey, date, exact || undefined);
     return cachedJson(rows);
   } catch (error) {
     console.error('Error fetching benchmarks:', error);

packages/app/src/components/inference/hooks/useChartData.ts

@@ -100,7 +100,9 @@ export function useChartData(
   }, [selectedGPUs, selectedDates, selectedDateRange, selectedRunDate]);
 
   const comparisonQueries = useQueries({
-    queries: comparisonDates.map((date) => benchmarkQueryOptions(selectedModel, date, enabled)),
+    queries: comparisonDates.map((date) =>
+      benchmarkQueryOptions(selectedModel, date, enabled, true),
+    ),
   });
 
   const comparisonLoading = comparisonQueries.some((q) => q.isLoading);

packages/app/src/hooks/api/use-benchmarks.ts

@@ -3,10 +3,15 @@ import { useQuery, keepPreviousData } from '@tanstack/react-query';
 import { fetchBenchmarks } from '@/lib/api';
 
 /** Shared query options — reused by useQueries for comparison dates. */
-export function benchmarkQueryOptions(model: string, date: string, enabled: boolean = true) {
+export function benchmarkQueryOptions(
+  model: string,
+  date: string,
+  enabled: boolean = true,
+  exact?: boolean,
+) {
   return {
-    queryKey: ['benchmarks', model, date] as const,
-    queryFn: () => fetchBenchmarks(model, date),
+    queryKey: ['benchmarks', model, date, exact ? 'exact' : 'latest'] as const,
+    queryFn: () => fetchBenchmarks(model, date, exact),
     enabled: enabled && Boolean(model),
   };
 }

packages/app/src/lib/api.ts

@@ -94,9 +94,10 @@ async function fetchJson<T>(url: string): Promise<T> {
   return res.json();
 }
 
-export function fetchBenchmarks(model: string, date?: string) {
+export function fetchBenchmarks(model: string, date?: string, exact?: boolean) {
   const params = new URLSearchParams({ model });
   if (date) params.set('date', date);
+  if (exact) params.set('exact', 'true');
   return fetchJson<BenchmarkRow[]>(`/api/v1/benchmarks?${params}`);
 }
 

packages/db/src/queries/benchmarks.ts

@@ -38,9 +38,13 @@ export async function getLatestBenchmarks(
   sql: NeonClient,
   modelKey: string,
   date?: string,
+  exact?: boolean,
 ): Promise<BenchmarkRow[]> {
   if (date) {
     // Date-filtered: use base table with DISTINCT ON (the view only has the absolute latest)
+    // exact=true: only return data from this exact date (for GPU comparison)
+    // exact=false (default): return latest data as of this date (for main chart)
+    const dateFilter = exact ? sql`br.date = ${date}::date` : sql`br.date <= ${date}::date`;
     const rows = await sql`
       SELECT DISTINCT ON (br.config_id, br.conc, br.isl, br.osl)
         c.hardware,
@@ -71,7 +75,7 @@ export async function getLatestBenchmarks(
       JOIN latest_workflow_runs wr ON wr.id = br.workflow_run_id
       WHERE c.model = ${modelKey}
         AND br.error IS NULL
-        AND br.date <= ${date}::date
+        AND ${dateFilter}
       ORDER BY br.config_id, br.conc, br.isl, br.osl, br.date DESC
     `;
     return rows as unknown as BenchmarkRow[];