1 //===- DirectXTargetMachine.cpp - DirectX Target Implementation -*- 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 /// \file
10 /// This file contains DirectX target initializer.
11 ///
12 //===----------------------------------------------------------------------===//
13 
14 #include "DirectXTargetMachine.h"
15 #include "DXILWriter/DXILWriterPass.h"
16 #include "DirectX.h"
17 #include "DirectXSubtarget.h"
18 #include "DirectXTargetTransformInfo.h"
19 #include "TargetInfo/DirectXTargetInfo.h"
20 #include "llvm/CodeGen/Passes.h"
21 #include "llvm/CodeGen/TargetPassConfig.h"
22 #include "llvm/IR/IRPrintingPasses.h"
23 #include "llvm/IR/LegacyPassManager.h"
24 #include "llvm/MC/SectionKind.h"
25 #include "llvm/MC/TargetRegistry.h"
26 #include "llvm/Support/CodeGen.h"
27 #include "llvm/Support/Compiler.h"
28 #include "llvm/Support/ErrorHandling.h"
29 #include "llvm/Target/TargetLoweringObjectFile.h"
30 
31 using namespace llvm;
32 
33 extern "C" LLVM_EXTERNAL_VISIBILITY void LLVMInitializeDirectXTarget() {
34   RegisterTargetMachine<DirectXTargetMachine> X(getTheDirectXTarget());
35   auto *PR = PassRegistry::getPassRegistry();
36   initializeDXILPrepareModulePass(*PR);
37   initializeDXILOpLoweringLegacyPass(*PR);
38 }
39 
40 class DXILTargetObjectFile : public TargetLoweringObjectFile {
41 public:
42   DXILTargetObjectFile() = default;
43 
44   MCSection *getExplicitSectionGlobal(const GlobalObject *GO, SectionKind Kind,
45                                       const TargetMachine &TM) const override {
46     llvm_unreachable("Not supported!");
47   }
48 
49 protected:
50   MCSection *SelectSectionForGlobal(const GlobalObject *GO, SectionKind Kind,
51                                     const TargetMachine &TM) const override {
52     llvm_unreachable("Not supported!");
53   }
54 };
55 
56 class DirectXPassConfig : public TargetPassConfig {
57 public:
58   DirectXPassConfig(DirectXTargetMachine &TM, PassManagerBase &PM)
59       : TargetPassConfig(TM, PM) {}
60 
61   DirectXTargetMachine &getDirectXTargetMachine() const {
62     return getTM<DirectXTargetMachine>();
63   }
64 
65   FunctionPass *createTargetRegisterAllocator(bool) override { return nullptr; }
66 };
67 
68 DirectXTargetMachine::DirectXTargetMachine(const Target &T, const Triple &TT,
69                                            StringRef CPU, StringRef FS,
70                                            const TargetOptions &Options,
71                                            Optional<Reloc::Model> RM,
72                                            Optional<CodeModel::Model> CM,
73                                            CodeGenOpt::Level OL, bool JIT)
74     : LLVMTargetMachine(T,
75                         "e-m:e-p:32:32-i1:32-i8:8-i16:16-i32:32-i64:64-f16:16-"
76                         "f32:32-f64:64-n8:16:32:64",
77                         TT, CPU, FS, Options, Reloc::Static, CodeModel::Small,
78                         OL),
79       TLOF(std::make_unique<DXILTargetObjectFile>()),
80       Subtarget(std::make_unique<DirectXSubtarget>(TT, CPU, FS, *this)) {}
81 
82 DirectXTargetMachine::~DirectXTargetMachine() {}
83 
84 bool DirectXTargetMachine::addPassesToEmitFile(
85     PassManagerBase &PM, raw_pwrite_stream &Out, raw_pwrite_stream *DwoOut,
86     CodeGenFileType FileType, bool DisableVerify,
87     MachineModuleInfoWrapperPass *MMIWP) {
88   PM.add(createDXILOpLoweringLegacyPass());
89   PM.add(createDXILPrepareModulePass());
90   switch (FileType) {
91   case CGFT_AssemblyFile:
92     PM.add(createPrintModulePass(Out, "", true));
93     break;
94   case CGFT_ObjectFile:
95     // TODO: Use MC Object streamer to write DXContainer
96     PM.add(createDXILWriterPass(Out));
97     break;
98   case CGFT_Null:
99     break;
100   }
101   return false;
102 }
103 
104 bool DirectXTargetMachine::addPassesToEmitMC(PassManagerBase &PM,
105                                              MCContext *&Ctx,
106                                              raw_pwrite_stream &Out,
107                                              bool DisableVerify) {
108   return true;
109 }
110 
111 TargetPassConfig *DirectXTargetMachine::createPassConfig(PassManagerBase &PM) {
112   return new DirectXPassConfig(*this, PM);
113 }
114 
115 const DirectXSubtarget *
116 DirectXTargetMachine::getSubtargetImpl(const Function &) const {
117   return Subtarget.get();
118 }
119 
120 TargetTransformInfo
121 DirectXTargetMachine::getTargetTransformInfo(const Function &F) const {
122   return TargetTransformInfo(DirectXTTIImpl(this, F));
123 }
124 
125 DirectXTargetLowering::DirectXTargetLowering(const DirectXTargetMachine &TM,
126                                              const DirectXSubtarget &STI)
127     : TargetLowering(TM) {}
128