1#include "clang/Basic/OffloadArch.h"
2
3#include "llvm/ADT/STLExtras.h"
4#include "llvm/ADT/StringRef.h"
5
6namespace clang {
7
8namespace {
9struct OffloadArchToStringMap {
10 OffloadArch Arch;
11 const char *ArchName;
12 const char *VirtualArchName;
13};
14} // namespace
15
16#define SM(sm) {OffloadArch::SM_##sm, "sm_" #sm, "compute_" #sm}
17#define GFX(gpu) {OffloadArch::GFX##gpu, "gfx" #gpu, "compute_amdgcn"}
18static const OffloadArchToStringMap ArchNames[] = {
19 // clang-format off
20 {.Arch: OffloadArch::UNUSED, .ArchName: "", .VirtualArchName: ""},
21 SM(20), {.Arch: OffloadArch::SM_21, .ArchName: "sm_21", .VirtualArchName: "compute_20"}, // Fermi
22 SM(30), {.Arch: OffloadArch::SM_32_, .ArchName: "sm_32", .VirtualArchName: "compute_32"}, SM(35), SM(37), // Kepler
23 SM(50), SM(52), SM(53), // Maxwell
24 SM(60), SM(61), SM(62), // Pascal
25 SM(70), SM(72), // Volta
26 SM(75), // Turing
27 SM(80), SM(86), // Ampere
28 SM(87), // Jetson/Drive AGX Orin
29 SM(88), // Ampere
30 SM(89), // Ada Lovelace
31 SM(90), // Hopper
32 SM(90a), // Hopper
33 SM(100), // Blackwell
34 SM(100a), // Blackwell
35 SM(101), // Blackwell
36 SM(101a), // Blackwell
37 SM(103), // Blackwell
38 SM(103a), // Blackwell
39 SM(110), // Blackwell
40 SM(110a), // Blackwell
41 SM(120), // Blackwell
42 SM(120a), // Blackwell
43 SM(121), // Blackwell
44 SM(121a), // Blackwell
45 GFX(600), // gfx600
46 GFX(601), // gfx601
47 GFX(602), // gfx602
48 GFX(700), // gfx700
49 GFX(701), // gfx701
50 GFX(702), // gfx702
51 GFX(703), // gfx703
52 GFX(704), // gfx704
53 GFX(705), // gfx705
54 GFX(801), // gfx801
55 GFX(802), // gfx802
56 GFX(803), // gfx803
57 GFX(805), // gfx805
58 GFX(810), // gfx810
59 {.Arch: OffloadArch::GFX9_GENERIC, .ArchName: "gfx9-generic", .VirtualArchName: "compute_amdgcn"},
60 GFX(900), // gfx900
61 GFX(902), // gfx902
62 GFX(904), // gfx903
63 GFX(906), // gfx906
64 GFX(908), // gfx908
65 GFX(909), // gfx909
66 GFX(90a), // gfx90a
67 GFX(90c), // gfx90c
68 {.Arch: OffloadArch::GFX9_4_GENERIC, .ArchName: "gfx9-4-generic", .VirtualArchName: "compute_amdgcn"},
69 GFX(942), // gfx942
70 GFX(950), // gfx950
71 {.Arch: OffloadArch::GFX10_1_GENERIC, .ArchName: "gfx10-1-generic", .VirtualArchName: "compute_amdgcn"},
72 GFX(1010), // gfx1010
73 GFX(1011), // gfx1011
74 GFX(1012), // gfx1012
75 GFX(1013), // gfx1013
76 {.Arch: OffloadArch::GFX10_3_GENERIC, .ArchName: "gfx10-3-generic", .VirtualArchName: "compute_amdgcn"},
77 GFX(1030), // gfx1030
78 GFX(1031), // gfx1031
79 GFX(1032), // gfx1032
80 GFX(1033), // gfx1033
81 GFX(1034), // gfx1034
82 GFX(1035), // gfx1035
83 GFX(1036), // gfx1036
84 {.Arch: OffloadArch::GFX11_GENERIC, .ArchName: "gfx11-generic", .VirtualArchName: "compute_amdgcn"},
85 GFX(1100), // gfx1100
86 GFX(1101), // gfx1101
87 GFX(1102), // gfx1102
88 GFX(1103), // gfx1103
89 GFX(1150), // gfx1150
90 GFX(1151), // gfx1151
91 GFX(1152), // gfx1152
92 GFX(1153), // gfx1153
93 {.Arch: OffloadArch::GFX12_GENERIC, .ArchName: "gfx12-generic", .VirtualArchName: "compute_amdgcn"},
94 GFX(1200), // gfx1200
95 GFX(1201), // gfx1201
96 GFX(1250), // gfx1250
97 GFX(1251), // gfx1251
98 GFX(1310), // gfx1310
99 {.Arch: OffloadArch::AMDGCNSPIRV, .ArchName: "amdgcnspirv", .VirtualArchName: "compute_amdgcn"},
100 // Intel CPUs
101 {.Arch: OffloadArch::GRANITERAPIDS, .ArchName: "graniterapids", .VirtualArchName: ""},
102 // Intel GPUS
103 {.Arch: OffloadArch::BMG_G21, .ArchName: "bmg_g21", .VirtualArchName: ""},
104 {.Arch: OffloadArch::Generic, .ArchName: "generic", .VirtualArchName: ""},
105 // clang-format on
106};
107#undef SM
108#undef GFX
109
110const char *OffloadArchToString(OffloadArch A) {
111 auto Result =
112 llvm::find_if(Range: ArchNames, P: [A](const OffloadArchToStringMap &Map) {
113 return A == Map.Arch;
114 });
115 if (Result == std::end(arr: ArchNames))
116 return "unknown";
117 return Result->ArchName;
118}
119
120const char *OffloadArchToVirtualArchString(OffloadArch A) {
121 auto Result =
122 llvm::find_if(Range: ArchNames, P: [A](const OffloadArchToStringMap &Map) {
123 return A == Map.Arch;
124 });
125 if (Result == std::end(arr: ArchNames))
126 return "unknown";
127 return Result->VirtualArchName;
128}
129
130OffloadArch StringToOffloadArch(llvm::StringRef S) {
131 auto Result =
132 llvm::find_if(Range: ArchNames, P: [S](const OffloadArchToStringMap &Map) {
133 return S == Map.ArchName;
134 });
135 if (Result == std::end(arr: ArchNames))
136 return OffloadArch::UNKNOWN;
137 return Result->Arch;
138}
139
140} // namespace clang
141