Skip to content

Commit

Permalink
Add cuda (a10g) device to TorchInductor dashboard (#5505)
Browse files Browse the repository at this point in the history
  • Loading branch information
huydhn authored Jul 26, 2024
1 parent 09ed774 commit 47609ff
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions torchci/components/benchmark/compilers/common.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -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",
};

0 comments on commit 47609ff

Please sign in to comment.