From d31df07d76517f3d131f8e1f9b7683e72745c8f0 Mon Sep 17 00:00:00 2001
From: Huy Do <huydhn@gmail.com>
Date: Wed, 5 Mar 2025 17:51:11 -0800
Subject: [PATCH] Show use torch.compile on TorchAO charts

---
 .../components/benchmark/llms/components/LLMsGraphPanel.tsx  | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/torchci/components/benchmark/llms/components/LLMsGraphPanel.tsx b/torchci/components/benchmark/llms/components/LLMsGraphPanel.tsx
index 614d820723..8e96e401e8 100644
--- a/torchci/components/benchmark/llms/components/LLMsGraphPanel.tsx
+++ b/torchci/components/benchmark/llms/components/LLMsGraphPanel.tsx
@@ -182,6 +182,11 @@ export default function LLMsGraphPanel({
               ) {
                 const isDynamic = record.extra!["is_dynamic"];
                 record.display = `${model} / ${isDynamic}`;
+              } else if (repoName === "pytorch/ao") {
+                let useCompile = record.extra!["use_torch_compile"];
+                record.display = `${model} (${dtype}${
+                  useCompile === "true" ? " / compile" : ""
+                })`;
               } else {
                 record.display = model.includes(dtype)
                   ? model.includes(device)