contrib/llvm/tools/clang/lib/Basic/Cuda.cpp

   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