1 #include "clang/Basic/Cuda.h"
2 
3 #include "llvm/ADT/StringRef.h"
4 #include "llvm/ADT/StringSwitch.h"
5 #include "llvm/Support/ErrorHandling.h"
6 
7 namespace clang {
8 
9 const char *CudaVersionToString(CudaVersion V) {
10   switch (V) {
11   case CudaVersion::UNKNOWN:
12     return "unknown";
13   case CudaVersion::CUDA_70:
14     return "7.0";
15   case CudaVersion::CUDA_75:
16     return "7.5";
17   case CudaVersion::CUDA_80:
18     return "8.0";
19   case CudaVersion::CUDA_90:
20     return "9.0";
21   case CudaVersion::CUDA_91:
22     return "9.1";
23   case CudaVersion::CUDA_92:
24     return "9.2";
25   }
26   llvm_unreachable("invalid enum");
27 }
28 
29 const char *CudaArchToString(CudaArch A) {
30   switch (A) {
31   case CudaArch::LAST:
32     break;
33   case CudaArch::UNKNOWN:
34     return "unknown";
35   case CudaArch::SM_20:
36     return "sm_20";
37   case CudaArch::SM_21:
38     return "sm_21";
39   case CudaArch::SM_30:
40     return "sm_30";
41   case CudaArch::SM_32:
42     return "sm_32";
43   case CudaArch::SM_35:
44     return "sm_35";
45   case CudaArch::SM_37:
46     return "sm_37";
47   case CudaArch::SM_50:
48     return "sm_50";
49   case CudaArch::SM_52:
50     return "sm_52";
51   case CudaArch::SM_53:
52     return "sm_53";
53   case CudaArch::SM_60:
54     return "sm_60";
55   case CudaArch::SM_61:
56     return "sm_61";
57   case CudaArch::SM_62:
58     return "sm_62";
59   case CudaArch::SM_70:
60     return "sm_70";
61   case CudaArch::SM_72:
62     return "sm_72";
63   case CudaArch::GFX600: // tahiti
64     return "gfx600";
65   case CudaArch::GFX601: // pitcairn, verde, oland,hainan
66     return "gfx601";
67   case CudaArch::GFX700: // kaveri
68     return "gfx700";
69   case CudaArch::GFX701: // hawaii
70     return "gfx701";
71   case CudaArch::GFX702: // 290,290x,R390,R390x
72     return "gfx702";
73   case CudaArch::GFX703: // kabini mullins
74     return "gfx703";
75   case CudaArch::GFX704: // bonaire
76     return "gfx704";
77   case CudaArch::GFX801: // carrizo
78     return "gfx801";
79   case CudaArch::GFX802: // tonga,iceland
80     return "gfx802";
81   case CudaArch::GFX803: // fiji,polaris10
82     return "gfx803";
83   case CudaArch::GFX810: // stoney
84     return "gfx810";
85   case CudaArch::GFX900: // vega, instinct
86     return "gfx900";
87   case CudaArch::GFX902: // TBA
88     return "gfx902";
89   }
90   llvm_unreachable("invalid enum");
91 }
92 
93 CudaArch StringToCudaArch(llvm::StringRef S) {
94   return llvm::StringSwitch<CudaArch>(S)
95       .Case("sm_20", CudaArch::SM_20)
96       .Case("sm_21", CudaArch::SM_21)
97       .Case("sm_30", CudaArch::SM_30)
98       .Case("sm_32", CudaArch::SM_32)
99       .Case("sm_35", CudaArch::SM_35)
100       .Case("sm_37", CudaArch::SM_37)
101       .Case("sm_50", CudaArch::SM_50)
102       .Case("sm_52", CudaArch::SM_52)
103       .Case("sm_53", CudaArch::SM_53)
104       .Case("sm_60", CudaArch::SM_60)
105       .Case("sm_61", CudaArch::SM_61)
106       .Case("sm_62", CudaArch::SM_62)
107       .Case("sm_70", CudaArch::SM_70)
108       .Case("sm_72", CudaArch::SM_72)
109       .Case("gfx600", CudaArch::GFX600)
110       .Case("gfx601", CudaArch::GFX601)
111       .Case("gfx700", CudaArch::GFX700)
112       .Case("gfx701", CudaArch::GFX701)
113       .Case("gfx702", CudaArch::GFX702)
114       .Case("gfx703", CudaArch::GFX703)
115       .Case("gfx704", CudaArch::GFX704)
116       .Case("gfx801", CudaArch::GFX801)
117       .Case("gfx802", CudaArch::GFX802)
118       .Case("gfx803", CudaArch::GFX803)
119       .Case("gfx810", CudaArch::GFX810)
120       .Case("gfx900", CudaArch::GFX900)
121       .Case("gfx902", CudaArch::GFX902)
122       .Default(CudaArch::UNKNOWN);
123 }
124 
125 const char *CudaVirtualArchToString(CudaVirtualArch A) {
126   switch (A) {
127   case CudaVirtualArch::UNKNOWN:
128     return "unknown";
129   case CudaVirtualArch::COMPUTE_20:
130     return "compute_20";
131   case CudaVirtualArch::COMPUTE_30:
132     return "compute_30";
133   case CudaVirtualArch::COMPUTE_32:
134     return "compute_32";
135   case CudaVirtualArch::COMPUTE_35:
136     return "compute_35";
137   case CudaVirtualArch::COMPUTE_37:
138     return "compute_37";
139   case CudaVirtualArch::COMPUTE_50:
140     return "compute_50";
141   case CudaVirtualArch::COMPUTE_52:
142     return "compute_52";
143   case CudaVirtualArch::COMPUTE_53:
144     return "compute_53";
145   case CudaVirtualArch::COMPUTE_60:
146     return "compute_60";
147   case CudaVirtualArch::COMPUTE_61:
148     return "compute_61";
149   case CudaVirtualArch::COMPUTE_62:
150     return "compute_62";
151   case CudaVirtualArch::COMPUTE_70:
152     return "compute_70";
153   case CudaVirtualArch::COMPUTE_72:
154     return "compute_72";
155   case CudaVirtualArch::COMPUTE_AMDGCN:
156     return "compute_amdgcn";
157   }
158   llvm_unreachable("invalid enum");
159 }
160 
161 CudaVirtualArch StringToCudaVirtualArch(llvm::StringRef S) {
162   return llvm::StringSwitch<CudaVirtualArch>(S)
163       .Case("compute_20", CudaVirtualArch::COMPUTE_20)
164       .Case("compute_30", CudaVirtualArch::COMPUTE_30)
165       .Case("compute_32", CudaVirtualArch::COMPUTE_32)
166       .Case("compute_35", CudaVirtualArch::COMPUTE_35)
167       .Case("compute_37", CudaVirtualArch::COMPUTE_37)
168       .Case("compute_50", CudaVirtualArch::COMPUTE_50)
169       .Case("compute_52", CudaVirtualArch::COMPUTE_52)
170       .Case("compute_53", CudaVirtualArch::COMPUTE_53)
171       .Case("compute_60", CudaVirtualArch::COMPUTE_60)
172       .Case("compute_61", CudaVirtualArch::COMPUTE_61)
173       .Case("compute_62", CudaVirtualArch::COMPUTE_62)
174       .Case("compute_70", CudaVirtualArch::COMPUTE_70)
175       .Case("compute_72", CudaVirtualArch::COMPUTE_72)
176       .Case("compute_amdgcn", CudaVirtualArch::COMPUTE_AMDGCN)
177       .Default(CudaVirtualArch::UNKNOWN);
178 }
179 
180 CudaVirtualArch VirtualArchForCudaArch(CudaArch A) {
181   switch (A) {
182   case CudaArch::LAST:
183     break;
184   case CudaArch::UNKNOWN:
185     return CudaVirtualArch::UNKNOWN;
186   case CudaArch::SM_20:
187   case CudaArch::SM_21:
188     return CudaVirtualArch::COMPUTE_20;
189   case CudaArch::SM_30:
190     return CudaVirtualArch::COMPUTE_30;
191   case CudaArch::SM_32:
192     return CudaVirtualArch::COMPUTE_32;
193   case CudaArch::SM_35:
194     return CudaVirtualArch::COMPUTE_35;
195   case CudaArch::SM_37:
196     return CudaVirtualArch::COMPUTE_37;
197   case CudaArch::SM_50:
198     return CudaVirtualArch::COMPUTE_50;
199   case CudaArch::SM_52:
200     return CudaVirtualArch::COMPUTE_52;
201   case CudaArch::SM_53:
202     return CudaVirtualArch::COMPUTE_53;
203   case CudaArch::SM_60:
204     return CudaVirtualArch::COMPUTE_60;
205   case CudaArch::SM_61:
206     return CudaVirtualArch::COMPUTE_61;
207   case CudaArch::SM_62:
208     return CudaVirtualArch::COMPUTE_62;
209   case CudaArch::SM_70:
210     return CudaVirtualArch::COMPUTE_70;
211   case CudaArch::SM_72:
212     return CudaVirtualArch::COMPUTE_72;
213   case CudaArch::GFX600:
214   case CudaArch::GFX601:
215   case CudaArch::GFX700:
216   case CudaArch::GFX701:
217   case CudaArch::GFX702:
218   case CudaArch::GFX703:
219   case CudaArch::GFX704:
220   case CudaArch::GFX801:
221   case CudaArch::GFX802:
222   case CudaArch::GFX803:
223   case CudaArch::GFX810:
224   case CudaArch::GFX900:
225   case CudaArch::GFX902:
226     return CudaVirtualArch::COMPUTE_AMDGCN;
227   }
228   llvm_unreachable("invalid enum");
229 }
230 
231 CudaVersion MinVersionForCudaArch(CudaArch A) {
232   switch (A) {
233   case CudaArch::LAST:
234     break;
235   case CudaArch::UNKNOWN:
236     return CudaVersion::UNKNOWN;
237   case CudaArch::SM_20:
238   case CudaArch::SM_21:
239   case CudaArch::SM_30:
240   case CudaArch::SM_32:
241   case CudaArch::SM_35:
242   case CudaArch::SM_37:
243   case CudaArch::SM_50:
244   case CudaArch::SM_52:
245   case CudaArch::SM_53:
246     return CudaVersion::CUDA_70;
247   case CudaArch::SM_60:
248   case CudaArch::SM_61:
249   case CudaArch::SM_62:
250     return CudaVersion::CUDA_80;
251   case CudaArch::SM_70:
252     return CudaVersion::CUDA_90;
253   case CudaArch::SM_72:
254     return CudaVersion::CUDA_91;
255   case CudaArch::GFX600:
256   case CudaArch::GFX601:
257   case CudaArch::GFX700:
258   case CudaArch::GFX701:
259   case CudaArch::GFX702:
260   case CudaArch::GFX703:
261   case CudaArch::GFX704:
262   case CudaArch::GFX801:
263   case CudaArch::GFX802:
264   case CudaArch::GFX803:
265   case CudaArch::GFX810:
266   case CudaArch::GFX900:
267   case CudaArch::GFX902:
268     return CudaVersion::CUDA_70;
269   }
270   llvm_unreachable("invalid enum");
271 }
272 
273 CudaVersion MaxVersionForCudaArch(CudaArch A) {
274   switch (A) {
275   case CudaArch::UNKNOWN:
276     return CudaVersion::UNKNOWN;
277   case CudaArch::SM_20:
278   case CudaArch::SM_21:
279   case CudaArch::GFX600:
280   case CudaArch::GFX601:
281   case CudaArch::GFX700:
282   case CudaArch::GFX701:
283   case CudaArch::GFX702:
284   case CudaArch::GFX703:
285   case CudaArch::GFX704:
286   case CudaArch::GFX801:
287   case CudaArch::GFX802:
288   case CudaArch::GFX803:
289   case CudaArch::GFX810:
290   case CudaArch::GFX900:
291   case CudaArch::GFX902:
292     return CudaVersion::CUDA_80;
293   default:
294     return CudaVersion::LATEST;
295   }
296 }
297 
298 } // namespace clang
299