1 //===-- SPIRVSubtarget.cpp - SPIR-V Subtarget Information ------*- C++ -*--===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements the SPIR-V specific subclass of TargetSubtargetInfo.
10 //
11 //===----------------------------------------------------------------------===//
12
13 #include "SPIRVSubtarget.h"
14 #include "SPIRV.h"
15 #include "SPIRVCommandLine.h"
16 #include "SPIRVGlobalRegistry.h"
17 #include "SPIRVLegalizerInfo.h"
18 #include "SPIRVRegisterBankInfo.h"
19 #include "SPIRVTargetMachine.h"
20 #include "llvm/MC/TargetRegistry.h"
21 #include "llvm/TargetParser/Host.h"
22
23 using namespace llvm;
24
25 #define DEBUG_TYPE "spirv-subtarget"
26
27 #define GET_SUBTARGETINFO_TARGET_DESC
28 #define GET_SUBTARGETINFO_CTOR
29 #include "SPIRVGenSubtargetInfo.inc"
30
31 static cl::opt<bool>
32 SPVTranslatorCompat("translator-compatibility-mode",
33 cl::desc("SPIR-V Translator compatibility mode"),
34 cl::Optional, cl::init(false));
35
36 static cl::opt<std::set<SPIRV::Extension::Extension>, false,
37 SPIRVExtensionsParser>
38 Extensions("spirv-ext",
39 cl::desc("Specify list of enabled SPIR-V extensions"));
40
41 // Compare version numbers, but allow 0 to mean unspecified.
isAtLeastVer(VersionTuple Target,VersionTuple VerToCompareTo)42 static bool isAtLeastVer(VersionTuple Target, VersionTuple VerToCompareTo) {
43 return Target.empty() || Target >= VerToCompareTo;
44 }
45
SPIRVSubtarget(const Triple & TT,const std::string & CPU,const std::string & FS,const SPIRVTargetMachine & TM)46 SPIRVSubtarget::SPIRVSubtarget(const Triple &TT, const std::string &CPU,
47 const std::string &FS,
48 const SPIRVTargetMachine &TM)
49 : SPIRVGenSubtargetInfo(TT, CPU, /*TuneCPU=*/CPU, FS),
50 PointerSize(TM.getPointerSizeInBits(/* AS= */ 0)), InstrInfo(),
51 FrameLowering(initSubtargetDependencies(CPU, FS)), TLInfo(TM, *this),
52 TargetTriple(TT) {
53 switch (TT.getSubArch()) {
54 case Triple::SPIRVSubArch_v10:
55 SPIRVVersion = VersionTuple(1, 0);
56 break;
57 case Triple::SPIRVSubArch_v11:
58 SPIRVVersion = VersionTuple(1, 1);
59 break;
60 case Triple::SPIRVSubArch_v12:
61 SPIRVVersion = VersionTuple(1, 2);
62 break;
63 case Triple::SPIRVSubArch_v13:
64 SPIRVVersion = VersionTuple(1, 3);
65 break;
66 case Triple::SPIRVSubArch_v14:
67 default:
68 SPIRVVersion = VersionTuple(1, 4);
69 break;
70 case Triple::SPIRVSubArch_v15:
71 SPIRVVersion = VersionTuple(1, 5);
72 break;
73 case Triple::SPIRVSubArch_v16:
74 SPIRVVersion = VersionTuple(1, 6);
75 break;
76 }
77 OpenCLVersion = VersionTuple(2, 2);
78
79 // The order of initialization is important.
80 initAvailableExtensions();
81 initAvailableExtInstSets();
82
83 GR = std::make_unique<SPIRVGlobalRegistry>(PointerSize);
84 CallLoweringInfo = std::make_unique<SPIRVCallLowering>(TLInfo, GR.get());
85 InlineAsmInfo = std::make_unique<SPIRVInlineAsmLowering>(TLInfo);
86 Legalizer = std::make_unique<SPIRVLegalizerInfo>(*this);
87 RegBankInfo = std::make_unique<SPIRVRegisterBankInfo>();
88 InstSelector.reset(
89 createSPIRVInstructionSelector(TM, *this, *RegBankInfo.get()));
90 }
91
initSubtargetDependencies(StringRef CPU,StringRef FS)92 SPIRVSubtarget &SPIRVSubtarget::initSubtargetDependencies(StringRef CPU,
93 StringRef FS) {
94 ParseSubtargetFeatures(CPU, /*TuneCPU=*/CPU, FS);
95 return *this;
96 }
97
canUseExtension(SPIRV::Extension::Extension E) const98 bool SPIRVSubtarget::canUseExtension(SPIRV::Extension::Extension E) const {
99 return AvailableExtensions.contains(E);
100 }
101
canUseExtInstSet(SPIRV::InstructionSet::InstructionSet E) const102 bool SPIRVSubtarget::canUseExtInstSet(
103 SPIRV::InstructionSet::InstructionSet E) const {
104 return AvailableExtInstSets.contains(E);
105 }
106
isAtLeastSPIRVVer(VersionTuple VerToCompareTo) const107 bool SPIRVSubtarget::isAtLeastSPIRVVer(VersionTuple VerToCompareTo) const {
108 return isAtLeastVer(SPIRVVersion, VerToCompareTo);
109 }
110
isAtLeastOpenCLVer(VersionTuple VerToCompareTo) const111 bool SPIRVSubtarget::isAtLeastOpenCLVer(VersionTuple VerToCompareTo) const {
112 if (!isOpenCLEnv())
113 return false;
114 return isAtLeastVer(OpenCLVersion, VerToCompareTo);
115 }
116
117 // If the SPIR-V version is >= 1.4 we can call OpPtrEqual and OpPtrNotEqual.
118 // In SPIR-V Translator compatibility mode this feature is not available.
canDirectlyComparePointers() const119 bool SPIRVSubtarget::canDirectlyComparePointers() const {
120 return !SPVTranslatorCompat && isAtLeastVer(SPIRVVersion, VersionTuple(1, 4));
121 }
122
initAvailableExtensions()123 void SPIRVSubtarget::initAvailableExtensions() {
124 AvailableExtensions.clear();
125 if (!isOpenCLEnv())
126 return;
127
128 AvailableExtensions.insert(Extensions.begin(), Extensions.end());
129 }
130
131 // TODO: use command line args for this rather than just defaults.
132 // Must have called initAvailableExtensions first.
initAvailableExtInstSets()133 void SPIRVSubtarget::initAvailableExtInstSets() {
134 AvailableExtInstSets.clear();
135 if (!isOpenCLEnv())
136 AvailableExtInstSets.insert(SPIRV::InstructionSet::GLSL_std_450);
137 else
138 AvailableExtInstSets.insert(SPIRV::InstructionSet::OpenCL_std);
139
140 // Handle extended instruction sets from extensions.
141 if (canUseExtension(
142 SPIRV::Extension::SPV_AMD_shader_trinary_minmax_extension)) {
143 AvailableExtInstSets.insert(
144 SPIRV::InstructionSet::SPV_AMD_shader_trinary_minmax);
145 }
146 }
147