Index: include/llvm/Support/TargetParser.h =================================================================== --- include/llvm/Support/TargetParser.h +++ include/llvm/Support/TargetParser.h @@ -268,6 +268,86 @@ } // namespace X86 +namespace AMDGPU { + +/// GPU kinds supported by the AMDGPU target. +enum GPUKind : uint32_t { + // Not specified processor. + GK_NONE = 0, + + // R600-based processors. + GK_R600 = 1, + GK_R630 = 2, + GK_RS880 = 3, + GK_RV670 = 4, + GK_RV710 = 5, + GK_RV730 = 6, + GK_RV770 = 7, + GK_CEDAR = 8, + GK_CYPRESS = 9, + GK_JUNIPER = 10, + GK_REDWOOD = 11, + GK_SUMO = 12, + GK_BARTS = 13, + GK_CAICOS = 14, + GK_CAYMAN = 15, + GK_TURKS = 16, + + GK_R600_FIRST = GK_R600, + GK_R600_LAST = GK_TURKS, + + // AMDGCN-based processors. + GK_GFX600 = 32, + GK_GFX601 = 33, + + GK_GFX700 = 40, + GK_GFX701 = 41, + GK_GFX702 = 42, + GK_GFX703 = 43, + GK_GFX704 = 44, + + GK_GFX801 = 50, + GK_GFX802 = 51, + GK_GFX803 = 52, + GK_GFX810 = 53, + + GK_GFX900 = 60, + GK_GFX902 = 61, + GK_GFX904 = 62, + GK_GFX906 = 63, + + GK_AMDGCN_FIRST = GK_GFX600, + GK_AMDGCN_LAST = GK_GFX906, +}; + +// This isn't comprehensive for now, just things that are needed from the +// frontend driver. +enum ArchFeatureKind : uint32_t { + FEATURE_NONE = 0, + + // These features only exist for r600, and are implied true for amdgcn. + FEATURE_FMA = 1 << 1, + FEATURE_LDEXP = 1 << 2, + FEATURE_FP64 = 1 << 3, + + // Common features. + FEATURE_FAST_FMA_F32 = 1 << 4, + FEATURE_FAST_DENORMAL_F32 = 1 << 5 +}; + +GPUKind parseArchAMDGCN(StringRef CPU); +GPUKind parseArchR600(StringRef CPU); +StringRef getArchNameAMDGCN(GPUKind AK); +StringRef getArchNameR600(GPUKind AK); +StringRef getCanonicalArchName(StringRef Arch); +unsigned getArchAttrAMDGCN(GPUKind AK); +unsigned getArchAttrR600(GPUKind AK); + +void fillValidArchListAMDGCN(SmallVectorImpl &Values); +void fillValidArchListR600(SmallVectorImpl &Values); + +} + } // namespace llvm #endif Index: lib/Support/TargetParser.cpp =================================================================== --- lib/Support/TargetParser.cpp +++ lib/Support/TargetParser.cpp @@ -14,6 +14,7 @@ #include "llvm/Support/ARMBuildAttributes.h" #include "llvm/Support/TargetParser.h" +#include "llvm/ADT/ArrayRef.h" #include "llvm/ADT/StringSwitch.h" #include "llvm/ADT/Twine.h" #include @@ -944,3 +945,149 @@ bool llvm::AArch64::isX18ReservedByDefault(const Triple &TT) { return TT.isOSDarwin() || TT.isOSFuchsia() || TT.isOSWindows(); } + +struct GPUInfo { + StringLiteral Name; + StringLiteral CanonicalName; + AMDGPU::GPUKind Kind; + unsigned Features; +}; + +using namespace AMDGPU; +static constexpr GPUInfo R600GPUs[26] = { + // Name Canonical Kind Features + // Name + // + {{"r600"}, {"r600"}, GK_R600, FEATURE_NONE }, + {{"rv630"}, {"r600"}, GK_R600, FEATURE_NONE }, + {{"rv635"}, {"r600"}, GK_R600, FEATURE_NONE }, + {{"r630"}, {"r630"}, GK_R630, FEATURE_NONE }, + {{"rs780"}, {"rs880"}, GK_RS880, FEATURE_NONE }, + {{"rs880"}, {"rs880"}, GK_RS880, FEATURE_NONE }, + {{"rv610"}, {"rs880"}, GK_RS880, FEATURE_NONE }, + {{"rv620"}, {"rs880"}, GK_RS880, FEATURE_NONE }, + {{"rv670"}, {"rv670"}, GK_RV670, FEATURE_NONE }, + {{"rv710"}, {"rv710"}, GK_RV710, FEATURE_NONE }, + {{"rv730"}, {"rv730"}, GK_RV730, FEATURE_NONE }, + {{"rv740"}, {"rv770"}, GK_RV770, FEATURE_NONE }, + {{"rv770"}, {"rv770"}, GK_RV770, FEATURE_NONE }, + {{"cedar"}, {"cedar"}, GK_CEDAR, FEATURE_NONE }, + {{"palm"}, {"cedar"}, GK_CEDAR, FEATURE_NONE }, + {{"cypress"}, {"cypress"}, GK_CYPRESS, FEATURE_FMA }, + {{"hemlock"}, {"cypress"}, GK_CYPRESS, FEATURE_FMA }, + {{"juniper"}, {"juniper"}, GK_JUNIPER, FEATURE_NONE }, + {{"redwood"}, {"redwood"}, GK_REDWOOD, FEATURE_NONE }, + {{"sumo"}, {"sumo"}, GK_SUMO, FEATURE_NONE }, + {{"sumo2"}, {"sumo"}, GK_SUMO, FEATURE_NONE }, + {{"barts"}, {"barts"}, GK_BARTS, FEATURE_NONE }, + {{"caicos"}, {"caicos"}, GK_CAICOS, FEATURE_NONE }, + {{"aruba"}, {"cayman"}, GK_CAYMAN, FEATURE_FMA }, + {{"cayman"}, {"cayman"}, GK_CAYMAN, FEATURE_FMA }, + {{"turks"}, {"turks"}, GK_TURKS, FEATURE_NONE } +}; + +// This table should be sorted by the value of GPUKind +// Don't bother listing the implicitly true features +static constexpr GPUInfo AMDGCNGPUs[32] = { + // Name Canonical Kind Features + // Name + {{"gfx600"}, {"gfx600"}, GK_GFX600, FEATURE_FAST_FMA_F32}, + {{"tahiti"}, {"gfx600"}, GK_GFX600, FEATURE_FAST_FMA_F32}, + {{"gfx601"}, {"gfx601"}, GK_GFX601, FEATURE_NONE}, + {{"hainan"}, {"gfx601"}, GK_GFX601, FEATURE_NONE}, + {{"oland"}, {"gfx601"}, GK_GFX601, FEATURE_NONE}, + {{"pitcairn"}, {"gfx601"}, GK_GFX601, FEATURE_NONE}, + {{"verde"}, {"gfx601"}, GK_GFX601, FEATURE_NONE}, + {{"gfx700"}, {"gfx700"}, GK_GFX700, FEATURE_NONE}, + {{"kaveri"}, {"gfx700"}, GK_GFX700, FEATURE_NONE}, + {{"gfx701"}, {"gfx701"}, GK_GFX701, FEATURE_FAST_FMA_F32}, + {{"hawaii"}, {"gfx701"}, GK_GFX701, FEATURE_FAST_FMA_F32}, + {{"gfx702"}, {"gfx702"}, GK_GFX702, FEATURE_FAST_FMA_F32}, + {{"gfx703"}, {"gfx703"}, GK_GFX703, FEATURE_NONE}, + {{"kabini"}, {"gfx703"}, GK_GFX703, FEATURE_NONE}, + {{"mullins"}, {"gfx703"}, GK_GFX703, FEATURE_NONE}, + {{"gfx704"}, {"gfx704"}, GK_GFX704, FEATURE_NONE}, + {{"bonaire"}, {"gfx704"}, GK_GFX704, FEATURE_NONE}, + {{"gfx801"}, {"gfx801"}, GK_GFX801, FEATURE_FAST_FMA_F32|FEATURE_FAST_DENORMAL_F32}, + {{"carrizo"}, {"gfx801"}, GK_GFX801, FEATURE_FAST_FMA_F32|FEATURE_FAST_DENORMAL_F32}, + {{"gfx802"}, {"gfx802"}, GK_GFX802, FEATURE_FAST_DENORMAL_F32}, + {{"iceland"}, {"gfx802"}, GK_GFX802, FEATURE_FAST_DENORMAL_F32}, + {{"tonga"}, {"gfx802"}, GK_GFX802, FEATURE_FAST_DENORMAL_F32}, + {{"gfx803"}, {"gfx803"}, GK_GFX803, FEATURE_FAST_DENORMAL_F32}, + {{"fiji"}, {"gfx803"}, GK_GFX803, FEATURE_FAST_DENORMAL_F32}, + {{"polaris10"}, {"gfx803"}, GK_GFX803, FEATURE_FAST_DENORMAL_F32}, + {{"polaris11"}, {"gfx803"}, GK_GFX803, FEATURE_FAST_DENORMAL_F32}, + {{"gfx810"}, {"gfx810"}, GK_GFX810, FEATURE_FAST_DENORMAL_F32}, + {{"stoney"}, {"gfx810"}, GK_GFX810, FEATURE_FAST_DENORMAL_F32}, + {{"gfx900"}, {"gfx900"}, GK_GFX900, FEATURE_FAST_FMA_F32|FEATURE_FAST_DENORMAL_F32}, + {{"gfx902"}, {"gfx902"}, GK_GFX902, FEATURE_FAST_FMA_F32|FEATURE_FAST_DENORMAL_F32}, + {{"gfx904"}, {"gfx904"}, GK_GFX904, FEATURE_FAST_FMA_F32|FEATURE_FAST_DENORMAL_F32}, + {{"gfx906"}, {"gfx906"}, GK_GFX906, FEATURE_FAST_FMA_F32|FEATURE_FAST_DENORMAL_F32}, +}; + +static const GPUInfo *getArchEntry(AMDGPU::GPUKind AK, + ArrayRef Table) { + GPUInfo Search = { "", "", AK, AMDGPU::FEATURE_NONE }; + + auto I = std::lower_bound(Table.begin(), Table.end(), Search, + [](const GPUInfo &A, const GPUInfo &B) { + return A.Kind < B.Kind; + }); + + if (I == Table.end()) + return nullptr; + return I; +} + +StringRef llvm::AMDGPU::getArchNameAMDGCN(GPUKind AK) { + if (const auto *Entry = getArchEntry(AK, AMDGCNGPUs)) + return Entry->CanonicalName; + return ""; +} + +StringRef llvm::AMDGPU::getArchNameR600(GPUKind AK) { + if (const auto *Entry = getArchEntry(AK, R600GPUs)) + return Entry->CanonicalName; + return ""; +} + +AMDGPU::GPUKind llvm::AMDGPU::parseArchAMDGCN(StringRef CPU) { + for (const auto C : AMDGCNGPUs) { + if (CPU == C.Name) + return C.Kind; + } + + return AMDGPU::GPUKind::GK_NONE; +} + +AMDGPU::GPUKind llvm::AMDGPU::parseArchR600(StringRef CPU) { + for (const auto C : R600GPUs) { + if (CPU == C.Name) + return C.Kind; + } + + return AMDGPU::GPUKind::GK_NONE; +} + +unsigned AMDGPU::getArchAttrAMDGCN(GPUKind AK) { + if (const auto *Entry = getArchEntry(AK, AMDGCNGPUs)) + return Entry->Features; + return FEATURE_NONE; +} + +unsigned AMDGPU::getArchAttrR600(GPUKind AK) { + if (const auto *Entry = getArchEntry(AK, R600GPUs)) + return Entry->Features; + return FEATURE_NONE; +} + +void AMDGPU::fillValidArchListAMDGCN(SmallVectorImpl &Values) { + // XXX: Should this only report unique canonical names? + for (const auto C : AMDGCNGPUs) + Values.push_back(C.Name); +} + +void AMDGPU::fillValidArchListR600(SmallVectorImpl &Values) { + for (const auto C : R600GPUs) + Values.push_back(C.Name); +}