Skip to content

Commit d82a6e2

Browse files
authored
Add cuda (h100) device to PT2 inductor dashboard (#6276)
This is going to be needed by pytorch/pytorch#146868 ### Testing https://torchci-git-fork-huydhn-add-h100-fbopensource.vercel.app/benchmark/compilers shows `cuda (h100)` device
1 parent ea895ab commit d82a6e2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchci/components/benchmark/compilers/common.tsx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,15 +62,15 @@ export const DEFAULT_DEVICE_NAME = "cuda (a100)";
6262
// the LLM micro-benchmark page is implemented
6363
export const DISPLAY_NAMES_TO_DEVICE_NAMES: { [k: string]: string } = {
6464
"cuda (a100)": "cuda",
65-
"cuda (a10g)": "cuda_a10g",
65+
"cuda (h100)": "cuda_h100",
6666
"cpu (x86)": "cpu_x86",
6767
"cpu (aarch64)": "cpu_aarch64",
6868
"rocm (mi300x)": "rocm",
6969
mps: "mps",
7070
};
7171
export const DISPLAY_NAMES_TO_WORKFLOW_NAMES: { [k: string]: string } = {
7272
"cuda (a100)": "inductor-A100-perf-nightly",
73-
"cuda (a10g)": "inductor-perf-nightly-A10g",
73+
"cuda (h100)": "inductor-perf-nightly-h100",
7474
"cpu (x86)": "inductor-perf-nightly-x86",
7575
"cpu (aarch64)": "inductor-perf-nightly-aarch64",
7676
rocm: "inductor-perf-nightly-rocm",

0 commit comments

Comments
 (0)