d401a99306
At the end of the codegen pipeline for DXIL we will emit the DXIL into a global variable in the Module annotated for the "DXIL" section. This will be used by the MCDXContainerStreamer to emit the DXIL into a DXContainer DXIL part. Other parts of the DXContainer will be constructed similarly by serializing their values into GlobalVariables. This will allow DXIL to flow into DXContainers through the normal MCStreamer flow used in the MC layer. Depends on D122270 Reviewed By: kuhar Differential Revision: https://reviews.llvm.org/D125334
101 lines
3.1 KiB
C++
101 lines
3.1 KiB
C++
//===- DXILWriterPass.cpp - Bitcode writing pass --------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// DXILWriterPass implementation.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "DXILWriterPass.h"
|
|
#include "DXILBitcodeWriter.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Analysis/ModuleSummaryAnalysis.h"
|
|
#include "llvm/IR/Constants.h"
|
|
#include "llvm/IR/GlobalVariable.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/PassManager.h"
|
|
#include "llvm/InitializePasses.h"
|
|
#include "llvm/Pass.h"
|
|
#include "llvm/Support/Alignment.h"
|
|
#include "llvm/Transforms/Utils/ModuleUtils.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::dxil;
|
|
|
|
namespace {
|
|
class WriteDXILPass : public llvm::ModulePass {
|
|
raw_ostream &OS; // raw_ostream to print on
|
|
|
|
public:
|
|
static char ID; // Pass identification, replacement for typeid
|
|
WriteDXILPass() : ModulePass(ID), OS(dbgs()) {
|
|
initializeWriteDXILPassPass(*PassRegistry::getPassRegistry());
|
|
}
|
|
|
|
explicit WriteDXILPass(raw_ostream &o) : ModulePass(ID), OS(o) {
|
|
initializeWriteDXILPassPass(*PassRegistry::getPassRegistry());
|
|
}
|
|
|
|
StringRef getPassName() const override { return "Bitcode Writer"; }
|
|
|
|
bool runOnModule(Module &M) override {
|
|
WriteDXILToFile(M, OS);
|
|
return false;
|
|
}
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override {
|
|
AU.setPreservesAll();
|
|
}
|
|
};
|
|
|
|
class EmbedDXILPass : public llvm::ModulePass {
|
|
public:
|
|
static char ID; // Pass identification, replacement for typeid
|
|
EmbedDXILPass() : ModulePass(ID) {
|
|
initializeEmbedDXILPassPass(*PassRegistry::getPassRegistry());
|
|
}
|
|
|
|
StringRef getPassName() const override { return "DXIL Embedder"; }
|
|
|
|
bool runOnModule(Module &M) override {
|
|
std::string Data;
|
|
llvm::raw_string_ostream OS(Data);
|
|
WriteDXILToFile(M, OS);
|
|
|
|
Constant *ModuleConstant =
|
|
ConstantDataArray::get(M.getContext(), arrayRefFromStringRef(Data));
|
|
auto *GV = new llvm::GlobalVariable(M, ModuleConstant->getType(), true,
|
|
GlobalValue::PrivateLinkage,
|
|
ModuleConstant, "dx.dxil");
|
|
GV->setSection("DXIL");
|
|
GV->setAlignment(Align(4));
|
|
appendToCompilerUsed(M, {GV});
|
|
return true;
|
|
}
|
|
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override {
|
|
AU.setPreservesAll();
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
char WriteDXILPass::ID = 0;
|
|
INITIALIZE_PASS_BEGIN(WriteDXILPass, "write-bitcode", "Write Bitcode", false,
|
|
true)
|
|
INITIALIZE_PASS_DEPENDENCY(ModuleSummaryIndexWrapperPass)
|
|
INITIALIZE_PASS_END(WriteDXILPass, "write-bitcode", "Write Bitcode", false,
|
|
true)
|
|
|
|
ModulePass *llvm::createDXILWriterPass(raw_ostream &Str) {
|
|
return new WriteDXILPass(Str);
|
|
}
|
|
|
|
char EmbedDXILPass::ID = 0;
|
|
INITIALIZE_PASS(EmbedDXILPass, "dxil-embed", "Embed DXIL", false, true)
|
|
|
|
ModulePass *llvm::createDXILEmbedderPass() { return new EmbedDXILPass(); }
|