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 "SPIRVGlobalRegistry.h"
16 #include "SPIRVLegalizerInfo.h"
17 #include "SPIRVRegisterBankInfo.h"
18 #include "SPIRVTargetMachine.h"
19 #include "llvm/MC/TargetRegistry.h"
20 #include "llvm/Support/Host.h"
21 
22 using namespace llvm;
23 
24 #define DEBUG_TYPE "spirv-subtarget"
25 
26 #define GET_SUBTARGETINFO_TARGET_DESC
27 #define GET_SUBTARGETINFO_CTOR
28 #include "SPIRVGenSubtargetInfo.inc"
29 
30 // Compare version numbers, but allow 0 to mean unspecified.
31 static bool isAtLeastVer(uint32_t Target, uint32_t VerToCompareTo) {
32   return Target == 0 || Target >= VerToCompareTo;
33 }
34 
35 static unsigned computePointerSize(const Triple &TT) {
36   const auto Arch = TT.getArch();
37   // TODO: unify this with pointers legalization.
38   assert(TT.isSPIRV());
39   return Arch == Triple::spirv32 ? 32 : 64;
40 }
41 
42 SPIRVSubtarget::SPIRVSubtarget(const Triple &TT, const std::string &CPU,
43                                const std::string &FS,
44                                const SPIRVTargetMachine &TM)
45     : SPIRVGenSubtargetInfo(TT, CPU, /*TuneCPU=*/CPU, FS),
46       PointerSize(computePointerSize(TT)), SPIRVVersion(0), OpenCLVersion(0),
47       InstrInfo(), FrameLowering(initSubtargetDependencies(CPU, FS)),
48       TLInfo(TM, *this) {
49   // The order of initialization is important.
50   initAvailableExtensions();
51   initAvailableExtInstSets();
52 
53   GR = std::make_unique<SPIRVGlobalRegistry>(PointerSize);
54   CallLoweringInfo = std::make_unique<SPIRVCallLowering>(TLInfo, GR.get());
55   Legalizer = std::make_unique<SPIRVLegalizerInfo>(*this);
56   RegBankInfo = std::make_unique<SPIRVRegisterBankInfo>();
57   InstSelector.reset(
58       createSPIRVInstructionSelector(TM, *this, *RegBankInfo.get()));
59 }
60 
61 SPIRVSubtarget &SPIRVSubtarget::initSubtargetDependencies(StringRef CPU,
62                                                           StringRef FS) {
63   ParseSubtargetFeatures(CPU, /*TuneCPU=*/CPU, FS);
64   if (SPIRVVersion == 0)
65     SPIRVVersion = 14;
66   if (OpenCLVersion == 0)
67     OpenCLVersion = 22;
68   return *this;
69 }
70 
71 bool SPIRVSubtarget::canUseExtension(SPIRV::Extension::Extension E) const {
72   return AvailableExtensions.contains(E);
73 }
74 
75 bool SPIRVSubtarget::canUseExtInstSet(
76     SPIRV::InstructionSet::InstructionSet E) const {
77   return AvailableExtInstSets.contains(E);
78 }
79 
80 bool SPIRVSubtarget::isAtLeastSPIRVVer(uint32_t VerToCompareTo) const {
81   return isAtLeastVer(SPIRVVersion, VerToCompareTo);
82 }
83 
84 bool SPIRVSubtarget::isAtLeastOpenCLVer(uint32_t VerToCompareTo) const {
85   return isAtLeastVer(OpenCLVersion, VerToCompareTo);
86 }
87 
88 // If the SPIR-V version is >= 1.4 we can call OpPtrEqual and OpPtrNotEqual.
89 bool SPIRVSubtarget::canDirectlyComparePointers() const {
90   return isAtLeastVer(SPIRVVersion, 14);
91 }
92 
93 // TODO: use command line args for this rather than defaults.
94 void SPIRVSubtarget::initAvailableExtensions() {
95   AvailableExtensions.clear();
96   if (!isOpenCLEnv())
97     return;
98   // A default extension for testing.
99   AvailableExtensions.insert(
100       SPIRV::Extension::SPV_KHR_no_integer_wrap_decoration);
101 }
102 
103 // TODO: use command line args for this rather than just defaults.
104 // Must have called initAvailableExtensions first.
105 void SPIRVSubtarget::initAvailableExtInstSets() {
106   AvailableExtInstSets.clear();
107   if (!isOpenCLEnv())
108     AvailableExtInstSets.insert(SPIRV::InstructionSet::GLSL_std_450);
109   else
110     AvailableExtInstSets.insert(SPIRV::InstructionSet::OpenCL_std);
111 
112   // Handle extended instruction sets from extensions.
113   if (canUseExtension(
114           SPIRV::Extension::SPV_AMD_shader_trinary_minmax_extension)) {
115     AvailableExtInstSets.insert(
116         SPIRV::InstructionSet::SPV_AMD_shader_trinary_minmax);
117   }
118 }
119