From 47609fff9a8364f1b66431493c84812e5cbab9ed Mon Sep 17 00:00:00 2001 From: Huy Do Date: Fri, 26 Jul 2024 06:10:29 -0700 Subject: [PATCH] Add cuda (a10g) device to TorchInductor dashboard (#5505) The corresponding UX update to show cuda (a10g) device after https://github.com/pytorch/pytorch/pull/131816 ### Testing https://torchci-git-fork-huydhn-perf-dashboard-add-30bec4-fbopensource.vercel.app/benchmark/compilers --- torchci/components/benchmark/compilers/common.tsx | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchci/components/benchmark/compilers/common.tsx b/torchci/components/benchmark/compilers/common.tsx index 2952c21e18..cd93b916a4 100644 --- a/torchci/components/benchmark/compilers/common.tsx +++ b/torchci/components/benchmark/compilers/common.tsx @@ -58,11 +58,13 @@ export const DEFAULT_DEVICE_NAME = "cuda (a100)"; // the LLM micro-benchmark page is implemented export const DISPLAY_NAMES_TO_DEVICE_NAMES: { [k: string]: string } = { "cuda (a100)": "cuda", + "cuda (a10g)": "cuda_a10g", "cpu (x86)": "cpu_x86", "cpu (aarch64)": "cpu_aarch64", }; export const DISPLAY_NAMES_TO_WORKFLOW_NAMES: { [k: string]: string } = { "cuda (a100)": "inductor-A100-perf-nightly", + "cuda (a10g)": "inductor-perf-nightly-A10g", "cpu (x86)": "inductor-perf-nightly-x86", "cpu (aarch64)": "inductor-perf-nightly-aarch64", };