Skip to content

[tritonbench] Add triton compile time dashboard #6513

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 11 commits into from
Apr 8, 2025
Merged

Conversation

xuzhao9
Copy link
Contributor

@xuzhao9 xuzhao9 commented Apr 8, 2025

Located at /tritonbench/compile_time.

Data source: Tritonbench nightly run: https://github.com/pytorch-labs/tritonbench/actions/workflows/compile-time.yaml

Test plan:

image

cc @Jokeren

@xuzhao9 xuzhao9 requested a review from huydhn April 8, 2025 02:51
Copy link

vercel bot commented Apr 8, 2025

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name Status Preview Updated (UTC)
torchci ✅ Ready (Inspect) Visit Preview Apr 8, 2025 8:48am

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Apr 8, 2025
Copy link

@github-advanced-security github-advanced-security bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lintrunner found more than 20 potential problems in the proposed changes. Check the Files changed tab for more details.

return (
<>
<Grid2 container spacing={2}>
<Grid2 size={{ xs: 12, lg: 6 }} height={GRAPH_ROW_HEIGHT}>
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If there is only one chart, you could set lg to 12 to display it in the whole horizontal space instead of half the screen I think

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We will start from one chart and add more subgraphs later, e.g., more workload suites and break-down of the compilation time by stages (e.g., TTIR, TTGIR compile time).

@@ -0,0 +1,47 @@
WITH results AS (
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

As this is a brand new dashboard, it's probably ok to keep these separate queries for flexibility. But I suspect that we can fold them into existing https://github.com/pytorch/test-infra/tree/main/torchci/clickhouse_queries/oss_ci_benchmark_llms. That's for a later PR if needed

cc @yangw-dev

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Right, we are open to converge into the existing queries if that is possible.

Copy link
Contributor

@huydhn huydhn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM! Thank you for getting the initial version of TritonBench dashboard ready.

@xuzhao9 xuzhao9 merged commit 972fc89 into main Apr 8, 2025
6 checks passed
@xuzhao9 xuzhao9 deleted the xz9/tritonbench branch April 8, 2025 16:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants