Skip to content

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchci/components/benchmark/compilers/common.tsx

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ export const DEFAULT_DEVICE_NAME = "cuda (h100)";
6161
export const DISPLAY_NAMES_TO_DEVICE_NAMES: { [k: string]: string } = {
6262
"cuda (a100)": "cuda",
6363
"cuda (h100)": "cuda",
64+
"cuda (b200)": "cuda",
6465
"cpu (x86_64)": "cpu",
6566
"cpu (x86_zen)": "cpu",
6667
"cpu (aarch64)": "cpu",
@@ -70,6 +71,7 @@ export const DISPLAY_NAMES_TO_DEVICE_NAMES: { [k: string]: string } = {
7071
export const DISPLAY_NAMES_TO_ARCH_NAMES: { [k: string]: string } = {
7172
"cuda (a100)": "a100",
7273
"cuda (h100)": "h100",
74+
"cuda (b200)": "b200",
7375
"cpu (x86_64)": "x86_64",
7476
"cpu (x86_zen)": "x86_zen",
7577
"cpu (aarch64)": "aarch64",
@@ -80,6 +82,7 @@ export const DISPLAY_NAMES_TO_ARCH_NAMES: { [k: string]: string } = {
8082
export const DISPLAY_NAMES_TO_WORKFLOW_NAMES: { [k: string]: string } = {
8183
"cuda (a100)": "inductor-A100-perf-nightly",
8284
"cuda (h100)": "inductor-perf-nightly-h100",
85+
"cuda (b200)": "inductor-perf-b200",
8386
"cpu (x86)": "inductor-perf-nightly-x86",
8487
"cpu (x86_zen)": "inductor-perf-nightly-x86-zen",
8588
"cpu (aarch64)": "inductor-perf-nightly-aarch64",

0 commit comments

Comments
 (0)