Skip to content

Commit aaf8c9a

Browse files
authored
Add XPU device to PT2 inductor dashboard (#7417)
This is to support ROCm benchmark at pytorch/pytorch#166289 cc: @huydhn @atalman
1 parent 10eaa20 commit aaf8c9a

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

torchci/components/benchmark/compilers/common.tsx

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ export const DISPLAY_NAMES_TO_DEVICE_NAMES: { [k: string]: string } = {
6767
"cpu (aarch64)": "cpu",
6868
"rocm (mi300x)": "rocm",
6969
mps: "mps",
70+
xpu: "xpu",
7071
};
7172
export const DISPLAY_NAMES_TO_ARCH_NAMES: { [k: string]: string } = {
7273
"cuda (a100)": "a100",
@@ -78,6 +79,7 @@ export const DISPLAY_NAMES_TO_ARCH_NAMES: { [k: string]: string } = {
7879
// TODO (huydhn): Figure out a way to get the GPU name for ROCm
7980
"rocm (mi300x)": "",
8081
mps: "",
82+
xpu: "",
8183
};
8284
export const DISPLAY_NAMES_TO_WORKFLOW_NAMES: { [k: string]: string } = {
8385
"cuda (a100)": "inductor-A100-perf-nightly",
@@ -88,6 +90,7 @@ export const DISPLAY_NAMES_TO_WORKFLOW_NAMES: { [k: string]: string } = {
8890
"cpu (aarch64)": "inductor-perf-nightly-aarch64",
8991
rocm: "inductor-perf-nightly-rocm",
9092
mps: "inductor-perf-nightly-macos",
93+
xpu: "inductor-perf-nightly-xpu",
9194
};
9295

9396
export const DEFAULT_HIGHLIGHT_KEY = "none";

0 commit comments

Comments
 (0)