Skip to content

Commit a9ec424

Browse files
authored
[EZ] Fix TorchInductor CH queries for A100 (#6938)
As reported by @aorenste where [a100 results were not loaded on the dashboard](https://hud.pytorch.org/benchmark/compilers?dashboard=torchinductor&startTime=Wed%2C%2009%20Jul%202025%2023%3A22%3A18%20GMT&stopTime=Wed%2C%2016%20Jul%202025%2023%3A22%3A18%20GMT&granularity=hour&mode=inference&dtype=bfloat16&deviceName=cuda%20(a100)&lBranch=main&lCommit=&rBranch=main&rCommit=). The bug here is in the SQL queries where `arch` is now set correctly to `a100` by HUD. On the other hand, this legacy field in the database is empty because `a100` was the only arch in the past and wasn't set explicitly. With the work that @yangw-dev plans to do, these queries would likely not staying around for much longer. So, I just opt for the easy fix of handling `a100` explicitly. ### Testing [a100](https://torchci-git-fork-huydhn-fix-a100-query-fbopensource.vercel.app/benchmark/compilers?dashboard=torchinductor&startTime=Wed%2C%2009%20Jul%202025%2023%3A59%3A25%20GMT&stopTime=Wed%2C%2016%20Jul%202025%2023%3A59%3A25%20GMT&granularity=hour&mode=inference&dtype=bfloat16&deviceName=cuda%20(a100)&lBranch=main&lCommit=&rBranch=main&rCommit=) shows up correctly now cc @aorenste Signed-off-by: Huy Do <[email protected]>
1 parent 456e399 commit a9ec424

File tree

2 files changed

+2
-2
lines changed
  • torchci/clickhouse_queries

2 files changed

+2
-2
lines changed

torchci/clickhouse_queries/compilers_benchmark_performance/query.sql

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ WHERE
7676
-- in 6 months
7777
AND (
7878
(
79-
{arch: String } = ''
79+
({arch: String } = '' OR {arch: String } = 'a100')
8080
AND output LIKE CONCAT(
8181
'%_',
8282
{dtype: String },

torchci/clickhouse_queries/compilers_benchmark_performance_branches/query.sql

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ WHERE
1414
-- in 6 months
1515
AND (
1616
(
17-
{arch: String } = ''
17+
({arch: String } = '' OR {arch: String } = 'a100')
1818
AND benchmark_extra_info['output'] LIKE CONCAT(
1919
'%_',
2020
{dtype: String },

0 commit comments

Comments
 (0)