1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216 |
- ///////////////////////////////////////////////////////////////////////////////
- // //
- // HLModule.cpp //
- // Copyright (C) Microsoft Corporation. All rights reserved. //
- // This file is distributed under the University of Illinois Open Source //
- // License. See LICENSE.TXT for details. //
- // //
- // HighLevel DX IR module. //
- // //
- ///////////////////////////////////////////////////////////////////////////////
- #include "dxc/HLSL/DxilOperations.h"
- #include "dxc/HLSL/DxilShaderModel.h"
- #include "dxc/HLSL/DxilCBuffer.h"
- #include "dxc/HLSL/HLModule.h"
- #include "dxc/HLSL/DxilTypeSystem.h"
- #include "dxc/HLSL/DxilRootSignature.h"
- #include "llvm/IR/Constants.h"
- #include "llvm/IR/Function.h"
- #include "llvm/IR/IRBuilder.h"
- #include "llvm/IR/LLVMContext.h"
- #include "llvm/IR/Metadata.h"
- #include "llvm/IR/Module.h"
- #include "llvm/IR/DebugInfo.h"
- #include "llvm/IR/DIBuilder.h"
- #include "llvm/Support/raw_ostream.h"
- using namespace llvm;
- using std::string;
- using std::vector;
- using std::unique_ptr;
- namespace hlsl {
- static void
- CreateSignatures(const ShaderModel *pSM,
- std::unique_ptr<DxilSignature> &InputSignature,
- std::unique_ptr<DxilSignature> &OutputSignature,
- std::unique_ptr<DxilSignature> &PatchConstantSignature,
- std::unique_ptr<RootSignatureHandle> &RootSignature) {
- DXIL::ShaderKind shaderKind = pSM->GetKind();
- InputSignature = std::make_unique<DxilSignature>(shaderKind, DxilSignature::Kind::Input);
- OutputSignature = std::make_unique<DxilSignature>(shaderKind, DxilSignature::Kind::Output);
- PatchConstantSignature = std::make_unique<DxilSignature>(shaderKind, DxilSignature::Kind::PatchConstant);
- RootSignature = std::make_unique<RootSignatureHandle>();
- }
- //------------------------------------------------------------------------------
- //
- // HLModule methods.
- //
- HLModule::HLModule(Module *pModule)
- : m_Ctx(pModule->getContext())
- , m_pModule(pModule)
- , m_pEntryFunc(nullptr)
- , m_EntryName("")
- , m_pSM(nullptr)
- , m_pOP(std::make_unique<OP>(pModule->getContext(), pModule))
- , m_pTypeSystem(std::make_unique<DxilTypeSystem>(pModule))
- , m_pMDHelper(std::make_unique<DxilMDHelper>(
- pModule, std::make_unique<HLExtraPropertyHelper>(pModule)))
- , m_pDebugInfoFinder(nullptr)
- , m_DxilMajor(1)
- , m_DxilMinor(0) {
- DXASSERT_NOMSG(m_pModule != nullptr);
- // Pin LLVM dump methods. TODO: make debug-only.
- void (__thiscall Module::*pfnModuleDump)() const = &Module::dump;
- void (__thiscall Type::*pfnTypeDump)() const = &Type::dump;
- m_pUnused = (char *)&pfnModuleDump - (char *)&pfnTypeDump;
- }
- HLModule::~HLModule() {
- }
- LLVMContext &HLModule::GetCtx() const { return m_Ctx; }
- Module *HLModule::GetModule() const { return m_pModule; }
- OP *HLModule::GetOP() const { return m_pOP.get(); }
- void HLModule::SetShaderModel(const ShaderModel *pSM) {
- DXASSERT(m_pSM == nullptr, "shader model must not change for the module");
- m_pSM = pSM;
- m_pMDHelper->SetShaderModel(m_pSM);
- CreateSignatures(m_pSM, m_InputSignature, m_OutputSignature, m_PatchConstantSignature, m_RootSignature);
- }
- const ShaderModel *HLModule::GetShaderModel() const {
- return m_pSM;
- }
- uint32_t HLOptions::GetHLOptionsRaw() const {
- union Cast {
- Cast(const HLOptions &options) {
- hlOptions = options;
- }
- HLOptions hlOptions;
- uint32_t rawData;
- };
- static_assert(sizeof(uint32_t) == sizeof(HLOptions),
- "size must match to make sure no undefined bits when cast");
- Cast rawCast(*this);
- return rawCast.rawData;
- }
- void HLOptions::SetHLOptionsRaw(uint32_t data) {
- union Cast {
- Cast(uint32_t data) {
- rawData = data;
- }
- HLOptions hlOptions;
- uint64_t rawData;
- };
- Cast rawCast(data);
- *this = rawCast.hlOptions;
- }
- void HLModule::SetHLOptions(HLOptions &opts) {
- m_Options = opts;
- }
- const HLOptions &HLModule::GetHLOptions() const {
- return m_Options;
- }
- Function *HLModule::GetEntryFunction() const {
- return m_pEntryFunc;
- }
- void HLModule::SetEntryFunction(Function *pEntryFunc) {
- m_pEntryFunc = pEntryFunc;
- }
- const string &HLModule::GetEntryFunctionName() const { return m_EntryName; }
- void HLModule::SetEntryFunctionName(const string &name) { m_EntryName = name; }
- template<typename T> unsigned
- HLModule::AddResource(vector<unique_ptr<T> > &Vec, unique_ptr<T> pRes) {
- DXASSERT_NOMSG((unsigned)Vec.size() < UINT_MAX);
- unsigned Id = (unsigned)Vec.size();
- Vec.emplace_back(std::move(pRes));
- return Id;
- }
- unsigned HLModule::AddCBuffer(unique_ptr<DxilCBuffer> pCBuffer) {
- return AddResource<DxilCBuffer>(m_CBuffers, std::move(pCBuffer));
- }
- DxilCBuffer &HLModule::GetCBuffer(unsigned idx) {
- return *m_CBuffers[idx];
- }
- const DxilCBuffer &HLModule::GetCBuffer(unsigned idx) const {
- return *m_CBuffers[idx];
- }
- const vector<unique_ptr<DxilCBuffer> > &HLModule::GetCBuffers() const {
- return m_CBuffers;
- }
- unsigned HLModule::AddSampler(unique_ptr<DxilSampler> pSampler) {
- return AddResource<DxilSampler>(m_Samplers, std::move(pSampler));
- }
- DxilSampler &HLModule::GetSampler(unsigned idx) {
- return *m_Samplers[idx];
- }
- const DxilSampler &HLModule::GetSampler(unsigned idx) const {
- return *m_Samplers[idx];
- }
- const vector<unique_ptr<DxilSampler> > &HLModule::GetSamplers() const {
- return m_Samplers;
- }
- unsigned HLModule::AddSRV(unique_ptr<HLResource> pSRV) {
- return AddResource<HLResource>(m_SRVs, std::move(pSRV));
- }
- HLResource &HLModule::GetSRV(unsigned idx) {
- return *m_SRVs[idx];
- }
- const HLResource &HLModule::GetSRV(unsigned idx) const {
- return *m_SRVs[idx];
- }
- const vector<unique_ptr<HLResource> > &HLModule::GetSRVs() const {
- return m_SRVs;
- }
- unsigned HLModule::AddUAV(unique_ptr<HLResource> pUAV) {
- return AddResource<HLResource>(m_UAVs, std::move(pUAV));
- }
- HLResource &HLModule::GetUAV(unsigned idx) {
- return *m_UAVs[idx];
- }
- const HLResource &HLModule::GetUAV(unsigned idx) const {
- return *m_UAVs[idx];
- }
- const vector<unique_ptr<HLResource> > &HLModule::GetUAVs() const {
- return m_UAVs;
- }
- void HLModule::RemoveFunction(llvm::Function *F) {
- DXASSERT_NOMSG(F != nullptr);
- m_HLFunctionPropsMap.erase(F);
- if (m_pTypeSystem.get()->GetFunctionAnnotation(F))
- m_pTypeSystem.get()->EraseFunctionAnnotation(F);
- }
- template <typename TResource>
- bool RemoveResource(std::vector<std::unique_ptr<TResource>> &vec,
- GlobalVariable *pVariable) {
- for (auto p = vec.begin(), e = vec.end(); p != e; ++p) {
- if ((*p)->GetGlobalSymbol() == pVariable) {
- p = vec.erase(p);
- // Update ID.
- for (e = vec.end();p != e; ++p) {
- unsigned ID = (*p)->GetID()-1;
- (*p)->SetID(ID);
- }
- return true;
- }
- }
- return false;
- }
- bool RemoveResource(std::vector<GlobalVariable *> &vec,
- llvm::GlobalVariable *pVariable) {
- for (auto p = vec.begin(), e = vec.end(); p != e; ++p) {
- if (*p == pVariable) {
- vec.erase(p);
- return true;
- }
- }
- return false;
- }
- void HLModule::RemoveGlobal(llvm::GlobalVariable *GV) {
- RemoveResources(&GV, 1);
- }
- void HLModule::RemoveResources(llvm::GlobalVariable **ppVariables,
- unsigned count) {
- DXASSERT_NOMSG(count == 0 || ppVariables != nullptr);
- unsigned resourcesRemoved = count;
- for (unsigned i = 0; i < count; ++i) {
- GlobalVariable *pVariable = ppVariables[i];
- // This could be considerably faster - check variable type to see which
- // resource type this is rather than scanning all lists, and look for
- // usage and removal patterns.
- if (RemoveResource(m_CBuffers, pVariable))
- continue;
- if (RemoveResource(m_SRVs, pVariable))
- continue;
- if (RemoveResource(m_UAVs, pVariable))
- continue;
- if (RemoveResource(m_Samplers, pVariable))
- continue;
- // TODO: do m_TGSMVariables and m_StreamOutputs need maintenance?
- --resourcesRemoved; // Global variable is not a resource?
- }
- }
- HLModule::tgsm_iterator HLModule::tgsm_begin() {
- return m_TGSMVariables.begin();
- }
- HLModule::tgsm_iterator HLModule::tgsm_end() {
- return m_TGSMVariables.end();
- }
- void HLModule::AddGroupSharedVariable(GlobalVariable *GV) {
- m_TGSMVariables.emplace_back(GV);
- }
- DxilSignature &HLModule::GetInputSignature() {
- return *m_InputSignature;
- }
- DxilSignature &HLModule::GetOutputSignature() {
- return *m_OutputSignature;
- }
- DxilSignature &HLModule::GetPatchConstantSignature() {
- return *m_PatchConstantSignature;
- }
- RootSignatureHandle &HLModule::GetRootSignature() {
- return *m_RootSignature;
- }
- DxilTypeSystem &HLModule::GetTypeSystem() {
- return *m_pTypeSystem;
- }
- DxilSignature *HLModule::ReleaseInputSignature() {
- return m_InputSignature.release();
- }
- DxilSignature *HLModule::ReleaseOutputSignature() {
- return m_OutputSignature.release();
- }
- DxilSignature *HLModule::ReleasePatchConstantSignature() {
- return m_PatchConstantSignature.release();
- }
- DxilTypeSystem *HLModule::ReleaseTypeSystem() {
- return m_pTypeSystem.release();
- }
- RootSignatureHandle *HLModule::ReleaseRootSignature() {
- return m_RootSignature.release();
- }
- void HLModule::EmitLLVMUsed() {
- if (m_LLVMUsed.empty())
- return;
- vector<llvm::Constant*> GVs;
- GVs.resize(m_LLVMUsed.size());
- for (size_t i = 0, e = m_LLVMUsed.size(); i != e; i++) {
- GVs[i] = ConstantExpr::getAddrSpaceCast(cast<llvm::Constant>(&*m_LLVMUsed[i]), Type::getInt8PtrTy(m_Ctx));
- }
- ArrayType *pATy = ArrayType::get(Type::getInt8PtrTy(m_Ctx), GVs.size());
- GlobalVariable *pGV = new GlobalVariable(*m_pModule, pATy, false,
- GlobalValue::AppendingLinkage,
- ConstantArray::get(pATy, GVs),
- "llvm.used");
- pGV->setSection("llvm.metadata");
- }
- vector<GlobalVariable* > &HLModule::GetLLVMUsed() {
- return m_LLVMUsed;
- }
- bool HLModule::HasHLFunctionProps(llvm::Function *F) {
- return m_HLFunctionPropsMap.find(F) != m_HLFunctionPropsMap.end();
- }
- HLFunctionProps &HLModule::GetHLFunctionProps(llvm::Function *F) {
- return *m_HLFunctionPropsMap[F];
- }
- void HLModule::AddHLFunctionProps(llvm::Function *F, std::unique_ptr<HLFunctionProps> &info) {
- DXASSERT(m_HLFunctionPropsMap.count(F) == 0, "F already in map, info will be overwritten");
- m_HLFunctionPropsMap[F] = std::move(info);
- }
- DxilFunctionAnnotation *HLModule::GetFunctionAnnotation(llvm::Function *F) {
- return m_pTypeSystem->GetFunctionAnnotation(F);
- }
- DxilFunctionAnnotation *HLModule::AddFunctionAnnotation(llvm::Function *F) {
- DXASSERT(m_pTypeSystem->GetFunctionAnnotation(F)==nullptr, "function annotation already exist");
- return m_pTypeSystem->AddFunctionAnnotation(F);
- }
- void HLModule::AddResourceTypeAnnotation(llvm::Type *Ty,
- DXIL::ResourceClass resClass,
- DXIL::ResourceKind kind) {
- if (m_ResTypeAnnotation.count(Ty) == 0) {
- m_ResTypeAnnotation.emplace(Ty, std::make_pair(resClass, kind));
- } else {
- DXASSERT(resClass == m_ResTypeAnnotation[Ty].first, "resClass mismatch");
- DXASSERT(kind == m_ResTypeAnnotation[Ty].second, "kind mismatch");
- }
- }
- DXIL::ResourceClass HLModule::GetResourceClass(llvm::Type *Ty) {
- if (m_ResTypeAnnotation.count(Ty) > 0) {
- return m_ResTypeAnnotation[Ty].first;
- } else {
- return DXIL::ResourceClass::Invalid;
- }
- }
- DXIL::ResourceKind HLModule::GetResourceKind(llvm::Type *Ty) {
- if (m_ResTypeAnnotation.count(Ty) > 0) {
- return m_ResTypeAnnotation[Ty].second;
- } else {
- return DXIL::ResourceKind::Invalid;
- }
- }
- static unsigned GetIntAt(MDTuple *tuple, unsigned idx) {
- return DxilMDHelper::ConstMDToUint32(tuple->getOperand(idx));
- }
- static unsigned GetFloatAt(MDTuple *tuple, unsigned idx) {
- return DxilMDHelper::ConstMDToFloat(tuple->getOperand(idx));
- }
- static const StringRef kHLDxilFunctionPropertiesMDName = "dx.fnprops";
- static const StringRef kHLDxilOptionsMDName = "dx.options";
- static const StringRef kHLDxilResourceTypeAnnotationMDName = "dx.resource.type.annotation";
- // DXIL metadata serialization/deserialization.
- void HLModule::EmitHLMetadata() {
- m_pMDHelper->EmitDxilVersion(m_DxilMajor, m_DxilMinor);
- m_pMDHelper->EmitDxilShaderModel(m_pSM);
- MDTuple *pMDSignatures = m_pMDHelper->EmitDxilSignatures(*m_InputSignature,
- *m_OutputSignature,
- *m_PatchConstantSignature);
- MDTuple *pMDResources = EmitHLResources();
- MDTuple *pMDProperties = EmitHLShaderProperties();
- m_pMDHelper->EmitDxilTypeSystem(GetTypeSystem(), m_LLVMUsed);
- EmitLLVMUsed();
- MDTuple *pEntry = m_pMDHelper->EmitDxilEntryPointTuple(GetEntryFunction(), m_EntryName, pMDSignatures, pMDResources, pMDProperties);
- vector<MDNode *> Entries;
- Entries.emplace_back(pEntry);
- m_pMDHelper->EmitDxilEntryPoints(Entries);
- {
- NamedMDNode * fnProps = m_pModule->getOrInsertNamedMetadata(kHLDxilFunctionPropertiesMDName);
- for (auto && pair : m_HLFunctionPropsMap) {
- const hlsl::HLFunctionProps * props = pair.second.get();
- Metadata *MDVals[30];
- std::fill(MDVals, MDVals + _countof(MDVals), nullptr);
- unsigned valIdx = 0;
- MDVals[valIdx++] = ValueAsMetadata::get(pair.first);
- switch (m_pSM->GetKind()) {
- case DXIL::ShaderKind::Compute:
- MDVals[valIdx++] = m_pMDHelper->Uint32ToConstMD(props->ShaderProps.CS.numThreads[0]);
- MDVals[valIdx++] = m_pMDHelper->Uint32ToConstMD(props->ShaderProps.CS.numThreads[1]);
- MDVals[valIdx++] = m_pMDHelper->Uint32ToConstMD(props->ShaderProps.CS.numThreads[2]);
- break;
- case DXIL::ShaderKind::Geometry:
- MDVals[valIdx++] = m_pMDHelper->Uint8ToConstMD((uint8_t)props->ShaderProps.GS.inputPrimitive);
- MDVals[valIdx++] = m_pMDHelper->Uint32ToConstMD(props->ShaderProps.GS.maxVertexCount);
- for (size_t i = 0; i < _countof(props->ShaderProps.GS.streamPrimitiveTopologies); ++i)
- MDVals[valIdx++] = m_pMDHelper->Uint8ToConstMD((uint8_t)props->ShaderProps.GS.streamPrimitiveTopologies[i]);
- break;
- case DXIL::ShaderKind::Hull:
- MDVals[valIdx++] = ValueAsMetadata::get(props->ShaderProps.HS.patchConstantFunc);
- MDVals[valIdx++] = m_pMDHelper->Uint8ToConstMD((uint8_t)props->ShaderProps.HS.domain);
- MDVals[valIdx++] = m_pMDHelper->Uint8ToConstMD((uint8_t)props->ShaderProps.HS.partition);
- MDVals[valIdx++] = m_pMDHelper->Uint8ToConstMD((uint8_t)props->ShaderProps.HS.outputPrimitive);
- MDVals[valIdx++] = m_pMDHelper->Uint32ToConstMD(props->ShaderProps.HS.inputControlPoints);
- MDVals[valIdx++] = m_pMDHelper->Uint32ToConstMD(props->ShaderProps.HS.outputControlPoints);
- MDVals[valIdx++] = m_pMDHelper->FloatToConstMD(props->ShaderProps.HS.maxTessFactor);
- break;
- case DXIL::ShaderKind::Domain:
- MDVals[valIdx++] = m_pMDHelper->Uint8ToConstMD((uint8_t)props->ShaderProps.DS.domain);
- MDVals[valIdx++] = m_pMDHelper->Uint32ToConstMD(props->ShaderProps.DS.inputControlPoints);
- break;
- case DXIL::ShaderKind::Pixel:
- MDVals[valIdx++] = m_pMDHelper->BoolToConstMD(props->ShaderProps.PS.EarlyDepthStencil);
- break;
- }
- fnProps->addOperand(MDTuple::get(m_Ctx, ArrayRef<llvm::Metadata *>(MDVals, valIdx)));
- }
- NamedMDNode * options = m_pModule->getOrInsertNamedMetadata(kHLDxilOptionsMDName);
- uint32_t hlOptions = m_Options.GetHLOptionsRaw();
- options->addOperand(MDNode::get(m_Ctx, m_pMDHelper->Uint32ToConstMD(hlOptions)));
- NamedMDNode * resTyAnnotations = m_pModule->getOrInsertNamedMetadata(kHLDxilResourceTypeAnnotationMDName);
- resTyAnnotations->addOperand(EmitResTyAnnotations());
- }
- }
- void HLModule::LoadHLMetadata() {
- m_pMDHelper->LoadDxilVersion(m_DxilMajor, m_DxilMinor);
- m_pMDHelper->LoadDxilShaderModel(m_pSM);
- CreateSignatures(m_pSM, m_InputSignature, m_OutputSignature, m_PatchConstantSignature, m_RootSignature);
- const llvm::NamedMDNode *pEntries = m_pMDHelper->GetDxilEntryPoints();
- Function *pEntryFunc;
- string EntryName;
- const llvm::MDOperand *pSignatures, *pResources, *pProperties;
- m_pMDHelper->GetDxilEntryPoint(pEntries->getOperand(0), pEntryFunc, EntryName, pSignatures, pResources, pProperties);
- SetEntryFunction(pEntryFunc);
- SetEntryFunctionName(EntryName);
- m_pMDHelper->LoadDxilSignatures(*pSignatures, *m_InputSignature,
- *m_OutputSignature, *m_PatchConstantSignature);
- LoadHLResources(*pResources);
- LoadHLShaderProperties(*pProperties);
- m_pMDHelper->LoadDxilTypeSystem(*m_pTypeSystem.get());
- {
- NamedMDNode * fnProps = m_pModule->getNamedMetadata(kHLDxilFunctionPropertiesMDName);
- size_t propIdx = 0;
- while (propIdx < fnProps->getNumOperands()) {
- MDTuple *pProps = dyn_cast<MDTuple>(fnProps->getOperand(propIdx++));
- std::unique_ptr<hlsl::HLFunctionProps> props = std::make_unique<hlsl::HLFunctionProps>();
- unsigned idx = 0;
- Function *F = dyn_cast<Function>(dyn_cast<ValueAsMetadata>(pProps->getOperand(idx++))->getValue());
- switch (m_pSM->GetKind()) {
- case DXIL::ShaderKind::Compute:
- props->ShaderProps.CS.numThreads[0] = GetIntAt(pProps, idx++);
- props->ShaderProps.CS.numThreads[1] = GetIntAt(pProps, idx++);
- props->ShaderProps.CS.numThreads[2] = GetIntAt(pProps, idx++);
- break;
- case DXIL::ShaderKind::Geometry:
- props->ShaderProps.GS.inputPrimitive = (DXIL::InputPrimitive)GetIntAt(pProps, idx++);
- props->ShaderProps.GS.maxVertexCount = GetIntAt(pProps, idx++);
- for (size_t i = 0; i < _countof(props->ShaderProps.GS.streamPrimitiveTopologies); ++i)
- props->ShaderProps.GS.streamPrimitiveTopologies[i] = (DXIL::PrimitiveTopology)GetIntAt(pProps, idx++);
- break;
- case DXIL::ShaderKind::Hull:
- props->ShaderProps.HS.patchConstantFunc = dyn_cast<Function>(dyn_cast<ValueAsMetadata>(pProps->getOperand(idx++))->getValue());
- props->ShaderProps.HS.domain = (DXIL::TessellatorDomain)GetIntAt(pProps, idx++);
- props->ShaderProps.HS.partition = (DXIL::TessellatorPartitioning)GetIntAt(pProps, idx++);
- props->ShaderProps.HS.outputPrimitive = (DXIL::TessellatorOutputPrimitive)GetIntAt(pProps, idx++);
- props->ShaderProps.HS.inputControlPoints = GetIntAt(pProps, idx++);
- props->ShaderProps.HS.outputControlPoints = GetIntAt(pProps, idx++);
- props->ShaderProps.HS.maxTessFactor = GetFloatAt(pProps, idx++);
- break;
- case DXIL::ShaderKind::Domain:
- props->ShaderProps.DS.domain = (DXIL::TessellatorDomain)GetIntAt(pProps, idx++);
- props->ShaderProps.DS.inputControlPoints = GetIntAt(pProps, idx++);
- break;
- case DXIL::ShaderKind::Pixel:
- props->ShaderProps.PS.EarlyDepthStencil = GetIntAt(pProps, idx++);
- break;
- }
- m_HLFunctionPropsMap[F] = std::move(props);
- }
- const NamedMDNode * options = m_pModule->getOrInsertNamedMetadata(kHLDxilOptionsMDName);
- const MDNode *MDOptions = options->getOperand(0);
- m_Options.SetHLOptionsRaw(DxilMDHelper::ConstMDToUint32(MDOptions->getOperand(0)));
- NamedMDNode * resTyAnnotations = m_pModule->getOrInsertNamedMetadata(kHLDxilResourceTypeAnnotationMDName);
- const MDNode *MDResTyAnnotations = resTyAnnotations->getOperand(0);
- if (MDResTyAnnotations->getNumOperands())
- LoadResTyAnnotations(MDResTyAnnotations->getOperand(0));
- }
- }
- void HLModule::ClearHLMetadata(llvm::Module &M) {
- Module::named_metadata_iterator
- b = M.named_metadata_begin(),
- e = M.named_metadata_end();
- SmallVector<NamedMDNode*, 8> nodes;
- for (; b != e; ++b) {
- StringRef name = b->getName();
- if (name == DxilMDHelper::kDxilVersionMDName ||
- name == DxilMDHelper::kDxilShaderModelMDName ||
- name == DxilMDHelper::kDxilEntryPointsMDName ||
- name == DxilMDHelper::kDxilResourcesMDName ||
- name == DxilMDHelper::kDxilTypeSystemMDName ||
- name == kHLDxilFunctionPropertiesMDName || // TODO: adjust to proper name
- name == kHLDxilResourceTypeAnnotationMDName ||
- name == kHLDxilOptionsMDName ||
- name.startswith(DxilMDHelper::kDxilTypeSystemHelperVariablePrefix)) {
- nodes.push_back(b);
- }
- }
- for (size_t i = 0; i < nodes.size(); ++i) {
- M.eraseNamedMetadata(nodes[i]);
- }
- }
- MDTuple *HLModule::EmitHLResources() {
- // Emit SRV records.
- MDTuple *pTupleSRVs = nullptr;
- if (!m_SRVs.empty()) {
- vector<Metadata *> MDVals;
- for (size_t i = 0; i < m_SRVs.size(); i++) {
- MDVals.emplace_back(m_pMDHelper->EmitDxilSRV(*m_SRVs[i]));
- }
- pTupleSRVs = MDNode::get(m_Ctx, MDVals);
- }
- // Emit UAV records.
- MDTuple *pTupleUAVs = nullptr;
- if (!m_UAVs.empty()) {
- vector<Metadata *> MDVals;
- for (size_t i = 0; i < m_UAVs.size(); i++) {
- MDVals.emplace_back(m_pMDHelper->EmitDxilUAV(*m_UAVs[i]));
- }
- pTupleUAVs = MDNode::get(m_Ctx, MDVals);
- }
- // Emit CBuffer records.
- MDTuple *pTupleCBuffers = nullptr;
- if (!m_CBuffers.empty()) {
- vector<Metadata *> MDVals;
- for (size_t i = 0; i < m_CBuffers.size(); i++) {
- MDVals.emplace_back(m_pMDHelper->EmitDxilCBuffer(*m_CBuffers[i]));
- }
- pTupleCBuffers = MDNode::get(m_Ctx, MDVals);
- }
- // Emit Sampler records.
- MDTuple *pTupleSamplers = nullptr;
- if (!m_Samplers.empty()) {
- vector<Metadata *> MDVals;
- for (size_t i = 0; i < m_Samplers.size(); i++) {
- MDVals.emplace_back(m_pMDHelper->EmitDxilSampler(*m_Samplers[i]));
- }
- pTupleSamplers = MDNode::get(m_Ctx, MDVals);
- }
- if (pTupleSRVs != nullptr || pTupleUAVs != nullptr || pTupleCBuffers != nullptr || pTupleSamplers != nullptr) {
- return m_pMDHelper->EmitDxilResourceTuple(pTupleSRVs, pTupleUAVs, pTupleCBuffers, pTupleSamplers);
- } else {
- return nullptr;
- }
- }
- void HLModule::LoadHLResources(const llvm::MDOperand &MDO) {
- const llvm::MDTuple *pSRVs, *pUAVs, *pCBuffers, *pSamplers;
- m_pMDHelper->GetDxilResources(MDO, pSRVs, pUAVs, pCBuffers, pSamplers);
- // Load SRV records.
- if (pSRVs != nullptr) {
- for (unsigned i = 0; i < pSRVs->getNumOperands(); i++) {
- unique_ptr<HLResource> pSRV(new HLResource);
- m_pMDHelper->LoadDxilSRV(pSRVs->getOperand(i), *pSRV);
- AddSRV(std::move(pSRV));
- }
- }
- // Load UAV records.
- if (pUAVs != nullptr) {
- for (unsigned i = 0; i < pUAVs->getNumOperands(); i++) {
- unique_ptr<HLResource> pUAV(new HLResource);
- m_pMDHelper->LoadDxilUAV(pUAVs->getOperand(i), *pUAV);
- AddUAV(std::move(pUAV));
- }
- }
- // Load CBuffer records.
- if (pCBuffers != nullptr) {
- for (unsigned i = 0; i < pCBuffers->getNumOperands(); i++) {
- unique_ptr<DxilCBuffer> pCB = std::make_unique<DxilCBuffer>();
- m_pMDHelper->LoadDxilCBuffer(pCBuffers->getOperand(i), *pCB);
- AddCBuffer(std::move(pCB));
- }
- }
- // Load Sampler records.
- if (pSamplers != nullptr) {
- for (unsigned i = 0; i < pSamplers->getNumOperands(); i++) {
- unique_ptr<DxilSampler> pSampler(new DxilSampler);
- m_pMDHelper->LoadDxilSampler(pSamplers->getOperand(i), *pSampler);
- AddSampler(std::move(pSampler));
- }
- }
- }
- llvm::MDTuple *HLModule::EmitResTyAnnotations() {
- vector<Metadata *> MDVals;
- for (auto &resAnnotation : m_ResTypeAnnotation) {
- Metadata *TyMeta =
- ValueAsMetadata::get(UndefValue::get(resAnnotation.first));
- MDVals.emplace_back(TyMeta);
- MDVals.emplace_back(m_pMDHelper->Uint32ToConstMD(
- static_cast<unsigned>(resAnnotation.second.first)));
- MDVals.emplace_back(m_pMDHelper->Uint32ToConstMD(
- static_cast<unsigned>(resAnnotation.second.second)));
- }
- return MDNode::get(m_Ctx, MDVals);
- }
- void HLModule::LoadResTyAnnotations(const llvm::MDOperand &MDO) {
- if (MDO.get() == nullptr)
- return;
- const MDTuple *pTupleMD = dyn_cast<MDTuple>(MDO.get());
- IFTBOOL(pTupleMD != nullptr, DXC_E_INCORRECT_DXIL_METADATA);
- IFTBOOL((pTupleMD->getNumOperands() & 0x3) == 0,
- DXC_E_INCORRECT_DXIL_METADATA);
- for (unsigned iNode = 0; iNode < pTupleMD->getNumOperands(); iNode += 3) {
- const MDOperand &MDTy = pTupleMD->getOperand(iNode);
- const MDOperand &MDClass = pTupleMD->getOperand(iNode + 1);
- const MDOperand &MDKind = pTupleMD->getOperand(iNode + 2);
- Type *Ty = m_pMDHelper->ValueMDToValue(MDTy)->getType();
- DXIL::ResourceClass resClass = static_cast<DXIL::ResourceClass>(
- DxilMDHelper::ConstMDToUint32(MDClass));
- DXIL::ResourceKind kind =
- static_cast<DXIL::ResourceKind>(DxilMDHelper::ConstMDToUint32(MDKind));
- AddResourceTypeAnnotation(Ty, resClass, kind);
- }
- }
- MDTuple *HLModule::EmitHLShaderProperties() {
- vector<Metadata *> MDVals;
- if (!m_RootSignature->IsEmpty()) {
- MDVals.emplace_back(m_pMDHelper->Uint32ToConstMD(DxilMDHelper::kDxilRootSignatureTag));
- MDVals.emplace_back(m_pMDHelper->EmitRootSignature(*m_RootSignature.get()));
- }
- return MDNode::get(m_Ctx, MDVals);
- }
- void HLModule::LoadHLShaderProperties(const MDOperand &MDO) {
- if (MDO.get() == nullptr)
- return;
- const MDTuple *pTupleMD = dyn_cast<MDTuple>(MDO.get());
- IFTBOOL(pTupleMD != nullptr, DXC_E_INCORRECT_DXIL_METADATA);
- IFTBOOL((pTupleMD->getNumOperands() & 0x1) == 0, DXC_E_INCORRECT_DXIL_METADATA);
- for (unsigned iNode = 0; iNode < pTupleMD->getNumOperands(); iNode += 2) {
- unsigned Tag = DxilMDHelper::ConstMDToUint32(pTupleMD->getOperand(iNode));
- const MDOperand &MDO = pTupleMD->getOperand(iNode + 1);
- IFTBOOL(MDO.get() != nullptr, DXC_E_INCORRECT_DXIL_METADATA);
- switch (Tag) {
- case DxilMDHelper::kDxilRootSignatureTag:
- m_pMDHelper->LoadRootSignature(MDO, *m_RootSignature.get());
- break;
- default:
- // Ignore other extended properties for now.
- break;
- }
- }
- }
- // TODO: Don't check names.
- bool HLModule::IsStreamOutputType(llvm::Type *Ty) {
- if (StructType *ST = dyn_cast<StructType>(Ty)) {
- if (ST->getName().startswith("class.PointStream"))
- return true;
- if (ST->getName().startswith("class.LineStream"))
- return true;
- if (ST->getName().startswith("class.TriangleStream"))
- return true;
- }
- return false;
- }
- bool HLModule::IsStreamOutputPtrType(llvm::Type *Ty) {
- if (!Ty->isPointerTy())
- return false;
- Ty = Ty->getPointerElementType();
- return IsStreamOutputType(Ty);
- }
- bool HLModule::IsHLSLObjectType(llvm::Type *Ty) {
- if (llvm::StructType *ST = dyn_cast<llvm::StructType>(Ty)) {
- StringRef name = ST->getName();
- if (name.startswith("dx.types.wave_t"))
- return true;
- if (name.endswith("_slice_type"))
- return false;
- name = name.ltrim("class.");
- name = name.ltrim("struct.");
- if (name == "SamplerState")
- return true;
- if (name == "SamplerComparisonState")
- return true;
- if (name.startswith("TriangleStream"))
- return true;
- if (name.startswith("PointStream"))
- return true;
- if (name.startswith("LineStream"))
- return true;
- if (name.startswith("AppendStructuredBuffer"))
- return true;
- if (name.startswith("ConsumeStructuredBuffer"))
- return true;
- if (name.startswith("ConstantBuffer"))
- return true;
- name = name.ltrim("RasterizerOrdered");
- name = name.ltrim("RW");
- if (name == "ByteAddressBuffer")
- return true;
- if (name.startswith("Buffer"))
- return true;
- if (name.startswith("StructuredBuffer"))
- return true;
- if (name.startswith("Texture1D"))
- return true;
- if (name.startswith("Texture1DArray"))
- return true;
- if (name.startswith("Texture2D"))
- return true;
- if (name.startswith("Texture2DArray"))
- return true;
- if (name.startswith("Texture3D"))
- return true;
- if (name.startswith("TextureCube"))
- return true;
- if (name.startswith("TextureCubeArray"))
- return true;
- if (name.startswith("Texture2DMS"))
- return true;
- if (name.startswith("Texture2DMSArray"))
- return true;
- }
- return false;
- }
- unsigned
- HLModule::GetLegacyCBufferFieldElementSize(DxilFieldAnnotation &fieldAnnotation,
- llvm::Type *Ty,
- DxilTypeSystem &typeSys) {
- while (isa<ArrayType>(Ty)) {
- Ty = Ty->getArrayElementType();
- }
- // Bytes.
- unsigned compSize = fieldAnnotation.GetCompType().Is64Bit()?8:4;
- unsigned fieldSize = compSize;
- if (Ty->isVectorTy()) {
- fieldSize *= Ty->getVectorNumElements();
- } else if (StructType *ST = dyn_cast<StructType>(Ty)) {
- DxilStructAnnotation *EltAnnotation = typeSys.GetStructAnnotation(ST);
- if (EltAnnotation) {
- fieldSize = EltAnnotation->GetCBufferSize();
- } else {
- // Calculate size when don't have annotation.
- if (fieldAnnotation.HasMatrixAnnotation()) {
- const DxilMatrixAnnotation &matAnnotation =
- fieldAnnotation.GetMatrixAnnotation();
- unsigned rows = matAnnotation.Rows;
- unsigned cols = matAnnotation.Cols;
- if (matAnnotation.Orientation == MatrixOrientation::ColumnMajor) {
- rows = cols;
- cols = matAnnotation.Rows;
- } else if (matAnnotation.Orientation != MatrixOrientation::RowMajor) {
- // Invalid matrix orientation.
- fieldSize = 0;
- }
- fieldSize = (rows - 1) * 16 + cols * 4;
- } else {
- // Cannot find struct annotation.
- fieldSize = 0;
- }
- }
- }
- return fieldSize;
- }
- bool HLModule::IsStaticGlobal(GlobalVariable *GV) {
- return GV->getLinkage() == GlobalValue::LinkageTypes::InternalLinkage &&
- GV->getType()->getPointerAddressSpace() == DXIL::kDefaultAddrSpace;
- }
- bool HLModule::IsSharedMemoryGlobal(llvm::GlobalVariable *GV) {
- return GV->getType()->getPointerAddressSpace() == DXIL::kTGSMAddrSpace;
- }
- void HLModule::GetParameterRowsAndCols(Type *Ty, unsigned &rows, unsigned &cols,
- DxilParameterAnnotation ¶mAnnotation) {
- if (Ty->isPointerTy())
- Ty = Ty->getPointerElementType();
- // For array input of HS, DS, GS,
- // we need to skip the first level which size is based on primitive type.
- DxilParamInputQual inputQual = paramAnnotation.GetParamInputQual();
- bool skipOneLevelArray = inputQual == DxilParamInputQual::InputPatch;
- skipOneLevelArray |= inputQual == DxilParamInputQual::OutputPatch;
- skipOneLevelArray |= inputQual == DxilParamInputQual::InputPrimitive;
- if (skipOneLevelArray) {
- if (Ty->isArrayTy())
- Ty = Ty->getArrayElementType();
- }
- unsigned arraySize = 1;
- while (Ty->isArrayTy()) {
- arraySize *= Ty->getArrayNumElements();
- Ty = Ty->getArrayElementType();
- }
- rows = 1;
- cols = 1;
- if (paramAnnotation.HasMatrixAnnotation()) {
- const DxilMatrixAnnotation &matrix = paramAnnotation.GetMatrixAnnotation();
- if (matrix.Orientation == MatrixOrientation::RowMajor) {
- rows = matrix.Rows;
- cols = matrix.Cols;
- } else {
- DXASSERT(matrix.Orientation == MatrixOrientation::ColumnMajor, "");
- cols = matrix.Rows;
- rows = matrix.Cols;
- }
- } else if (Ty->isVectorTy())
- cols = Ty->getVectorNumElements();
- rows *= arraySize;
- }
- // For legacy data layout, everything less than 32 align to 32.
- static const StringRef kLegacyLayoutString = "e-m:e-p:32:32-i1:32:32-i8:32:32-i16:32:32-i64:64-f16:32-f80:32-n8:16:32-a:0:32-S32";
- const char *HLModule::GetLegacyDataLayoutDesc() {
- return kLegacyLayoutString.data();
- }
- template<typename BuilderTy>
- Value *HLModule::EmitHLOperationCall(BuilderTy &Builder,
- HLOpcodeGroup group, unsigned opcode,
- Type *RetType,
- ArrayRef<Value *> paramList,
- llvm::Module &M) {
- SmallVector<llvm::Type *, 4> paramTyList;
- // Add the opcode param
- llvm::Type *opcodeTy = llvm::Type::getInt32Ty(M.getContext());
- paramTyList.emplace_back(opcodeTy);
- for (Value *param : paramList) {
- paramTyList.emplace_back(param->getType());
- }
- llvm::FunctionType *funcTy =
- llvm::FunctionType::get(RetType, paramTyList, false);
- Function *opFunc = GetOrCreateHLFunction(M, funcTy, group, opcode);
- SmallVector<Value *, 4> opcodeParamList;
- Value *opcodeConst = Constant::getIntegerValue(opcodeTy, APInt(32, opcode));
- opcodeParamList.emplace_back(opcodeConst);
- opcodeParamList.append(paramList.begin(), paramList.end());
- return Builder.CreateCall(opFunc, opcodeParamList);
- }
- template
- Value *HLModule::EmitHLOperationCall(IRBuilder<> &Builder,
- HLOpcodeGroup group, unsigned opcode,
- Type *RetType,
- ArrayRef<Value *> paramList,
- llvm::Module &M);
- unsigned HLModule::FindCastOp(bool fromUnsigned, bool toUnsigned,
- llvm::Type *SrcTy, llvm::Type *DstTy) {
- Instruction::CastOps castOp = llvm::Instruction::CastOps::BitCast;
- if (SrcTy->isAggregateType() || DstTy->isAggregateType())
- return llvm::Instruction::CastOps::BitCast;
- uint32_t SrcBitSize = SrcTy->getScalarSizeInBits();
- uint32_t DstBitSize = DstTy->getScalarSizeInBits();
- if (SrcTy->isIntOrIntVectorTy() && DstTy->isIntOrIntVectorTy()) {
- if (SrcBitSize > DstBitSize)
- return Instruction::Trunc;
- if (toUnsigned)
- return Instruction::ZExt;
- else
- return Instruction::SExt;
- }
- if (SrcTy->isFPOrFPVectorTy() && DstTy->isFPOrFPVectorTy()) {
- if (SrcBitSize > DstBitSize)
- return Instruction::FPTrunc;
- else
- return Instruction::FPExt;
- }
- if (SrcTy->isIntOrIntVectorTy() && DstTy->isFPOrFPVectorTy())
- if (fromUnsigned)
- return Instruction::UIToFP;
- else
- return Instruction::SIToFP;
- if (SrcTy->isFPOrFPVectorTy() && DstTy->isIntOrIntVectorTy())
- if (toUnsigned)
- return Instruction::FPToUI;
- else
- return Instruction::FPToSI;
- DXASSERT_NOMSG(0);
- return castOp;
- }
- bool HLModule::HasPreciseAttributeWithMetadata(Instruction *I) {
- MDNode *preciseNode =
- I->getMetadata(DxilMDHelper::kDxilPreciseAttributeMDName);
- return preciseNode != nullptr;
- }
- void HLModule::MarkPreciseAttributeWithMetadata(Instruction *I) {
- LLVMContext &Ctx = I->getContext();
- MDNode *preciseNode = MDNode::get(
- Ctx, {MDString::get(Ctx, DxilMDHelper::kDxilPreciseAttributeMDName)});
- I->setMetadata(DxilMDHelper::kDxilPreciseAttributeMDName, preciseNode);
- }
- void HLModule::ClearPreciseAttributeWithMetadata(Instruction *I) {
- I->setMetadata(DxilMDHelper::kDxilPreciseAttributeMDName, nullptr);
- }
- static void MarkPreciseAttribute(Function *F) {
- LLVMContext &Ctx = F->getContext();
- MDNode *preciseNode = MDNode::get(
- Ctx, {MDString::get(Ctx, DxilMDHelper::kDxilPreciseAttributeMDName)});
- F->setMetadata(DxilMDHelper::kDxilPreciseAttributeMDName, preciseNode);
- }
- static void MarkPreciseAttributeOnValWithFunctionCall(
- llvm::Value *V, llvm::IRBuilder<> &Builder, llvm::Module &M) {
- Type *Ty = V->getType();
- Type *EltTy = Ty->getScalarType();
- // TODO: Only do this on basic types.
-
- FunctionType *preciseFuncTy =
- FunctionType::get(Type::getVoidTy(M.getContext()), {EltTy}, false);
- // The function will be deleted after precise propagate.
- std::string preciseFuncName = "dx.attribute.precise.";
- raw_string_ostream mangledNameStr(preciseFuncName);
- EltTy->print(mangledNameStr);
- mangledNameStr.flush();
- Function *preciseFunc =
- cast<Function>(M.getOrInsertFunction(preciseFuncName, preciseFuncTy));
- if (!HLModule::HasPreciseAttribute(preciseFunc))
- MarkPreciseAttribute(preciseFunc);
- if (Ty->isVectorTy()) {
- for (unsigned i = 0; i < Ty->getVectorNumElements(); i++) {
- Value *Elt = Builder.CreateExtractElement(V, i);
- Builder.CreateCall(preciseFunc, {Elt});
- }
- } else
- Builder.CreateCall(preciseFunc, {V});
- }
- void HLModule::MarkPreciseAttributeOnPtrWithFunctionCall(llvm::Value *Ptr,
- llvm::Module &M) {
- for (User *U : Ptr->users()) {
- // Skip load inst.
- if (LoadInst *LI = dyn_cast<LoadInst>(U))
- continue;
- if (StoreInst *SI = dyn_cast<StoreInst>(U)) {
- Value *V = SI->getValueOperand();
- if (isa<Instruction>(V)) {
- // Mark the Value with function call.
- IRBuilder<> Builder(SI);
- MarkPreciseAttributeOnValWithFunctionCall(V, Builder, M);
- }
- } else if (CallInst *CI = dyn_cast<CallInst>(U)) {
- if (CI->getType()->isVoidTy()) {
- IRBuilder<> Builder(CI);
- // For void type, cannot use as function arg.
- // Mark all arg for it?
- for (auto &arg : CI->arg_operands()) {
- MarkPreciseAttributeOnValWithFunctionCall(arg, Builder, M);
- }
- } else {
- IRBuilder<> Builder(CI->getNextNode());
- MarkPreciseAttributeOnValWithFunctionCall(CI, Builder, M);
- }
- } else {
- // Must be GEP here.
- GetElementPtrInst *GEP = cast<GetElementPtrInst>(U);
- MarkPreciseAttributeOnPtrWithFunctionCall(GEP, M);
- }
- }
- }
- bool HLModule::HasPreciseAttribute(Function *F) {
- MDNode *preciseNode =
- F->getMetadata(DxilMDHelper::kDxilPreciseAttributeMDName);
- return preciseNode != nullptr;
- }
- DIGlobalVariable *
- HLModule::FindGlobalVariableDebugInfo(GlobalVariable *GV,
- DebugInfoFinder &DbgInfoFinder) {
- struct GlobalFinder {
- GlobalVariable *GV;
- bool operator()(llvm::DIGlobalVariable *const arg) const {
- return arg->getVariable() == GV;
- }
- };
- GlobalFinder F = {GV};
- DebugInfoFinder::global_variable_iterator Found =
- std::find_if(DbgInfoFinder.global_variables().begin(),
- DbgInfoFinder.global_variables().end(), F);
- if (Found != DbgInfoFinder.global_variables().end()) {
- return *Found;
- }
- return nullptr;
- }
- static void AddDIGlobalVariable(DIBuilder &Builder, DIGlobalVariable *LocDIGV,
- StringRef Name, DIType *DITy,
- GlobalVariable *GV, DebugInfoFinder &DbgInfoFinder, bool removeLocDIGV) {
- DIGlobalVariable *EltDIGV = Builder.createGlobalVariable(
- LocDIGV->getScope(), Name, GV->getName(), LocDIGV->getFile(),
- LocDIGV->getLine(), DITy, false, GV);
- DICompileUnit *DICU = dyn_cast<DICompileUnit>(LocDIGV->getScope());
- if (!DICU) {
- DISubprogram *DIS = dyn_cast<DISubprogram>(LocDIGV->getScope());
- if (DIS) {
- // Find the DICU which has this Subprogram.
- NamedMDNode *CompileUnits = GV->getParent()->getNamedMetadata("llvm.dbg.cu");
- DXASSERT_NOMSG(CompileUnits);
- for (unsigned I = 0, E = CompileUnits->getNumOperands(); I != E; ++I) {
- auto *CU = cast<DICompileUnit>(CompileUnits->getOperand(I));
- DXASSERT(CU , "Expected valid compile unit");
- for (DISubprogram *SP : CU->getSubprograms()) {
- if (SP == DIS) {
- DICU = CU;
- break;
- }
- }
- }
- }
- }
- DXASSERT_NOMSG(DICU);
- // Add global to CU.
- auto *GlobalVariables = DICU->getRawGlobalVariables();
- DXASSERT_NOMSG(GlobalVariables);
- MDTuple *GTuple = cast<MDTuple>(GlobalVariables);
- std::vector<Metadata *> AllGVs(GTuple->operands().begin(),
- GTuple->operands().end());
- if (removeLocDIGV) {
- auto locIt = std::find(AllGVs.begin(), AllGVs.end(), LocDIGV);
- AllGVs.erase(locIt);
- }
- AllGVs.emplace_back(EltDIGV);
- DICU->replaceGlobalVariables(MDTuple::get(GV->getContext(), AllGVs));
- DXVERIFY_NOMSG(DbgInfoFinder.appendGlobalVariable(EltDIGV));
- }
- void HLModule::CreateElementGlobalVariableDebugInfo(
- GlobalVariable *GV, DebugInfoFinder &DbgInfoFinder, GlobalVariable *EltGV,
- unsigned sizeInBits, unsigned alignInBits, unsigned offsetInBits,
- StringRef eltName) {
- DIGlobalVariable *DIGV = FindGlobalVariableDebugInfo(GV, DbgInfoFinder);
- DXASSERT_NOMSG(DIGV);
- DIBuilder Builder(*GV->getParent());
- DITypeIdentifierMap EmptyMap;
- DIType *DITy = DIGV->getType().resolve(EmptyMap);
- DIScope *DITyScope = DITy->getScope().resolve(EmptyMap);
- // Create Elt type.
- DIType *EltDITy =
- Builder.createMemberType(DITyScope, DITy->getName().str() + eltName.str(),
- DITy->getFile(), DITy->getLine(), sizeInBits,
- alignInBits, offsetInBits, /*Flags*/ 0, DITy);
- AddDIGlobalVariable(Builder, DIGV, DIGV->getName().str() + eltName.str(),
- EltDITy, EltGV, DbgInfoFinder, /*removeDIGV*/false);
- }
- void HLModule::UpdateGlobalVariableDebugInfo(
- llvm::GlobalVariable *GV, llvm::DebugInfoFinder &DbgInfoFinder,
- llvm::GlobalVariable *NewGV) {
- DIGlobalVariable *DIGV = FindGlobalVariableDebugInfo(GV, DbgInfoFinder);
- DXASSERT_NOMSG(DIGV);
- DIBuilder Builder(*GV->getParent());
- DITypeIdentifierMap EmptyMap;
- DIType *DITy = DIGV->getType().resolve(EmptyMap);
- AddDIGlobalVariable(Builder, DIGV, DIGV->getName(), DITy, NewGV,
- DbgInfoFinder,/*removeDIGV*/true);
- }
- DebugInfoFinder &HLModule::GetOrCreateDebugInfoFinder() {
- if (m_pDebugInfoFinder == nullptr) {
- m_pDebugInfoFinder = std::make_unique<llvm::DebugInfoFinder>();
- m_pDebugInfoFinder->processModule(*m_pModule);
- }
- return *m_pDebugInfoFinder;
- }
- //------------------------------------------------------------------------------
- //
- // Signature methods.
- //
- HLExtraPropertyHelper::HLExtraPropertyHelper(llvm::Module *pModule)
- : DxilExtraPropertyHelper(pModule) {
- }
- void HLExtraPropertyHelper::EmitSignatureElementProperties(const DxilSignatureElement &SE,
- vector<Metadata *> &MDVals) {
- }
- void HLExtraPropertyHelper::LoadSignatureElementProperties(const MDOperand &MDO,
- DxilSignatureElement &SE) {
- if (MDO.get() == nullptr)
- return;
- }
- } // namespace hlsl
- namespace llvm {
- hlsl::HLModule &Module::GetOrCreateHLModule(bool skipInit) {
- std::unique_ptr<hlsl::HLModule> M;
- if (!HasHLModule()) {
- M = std::make_unique<hlsl::HLModule>(this);
- if (!skipInit) {
- M->LoadHLMetadata();
- }
- SetHLModule(M.release());
- }
- return GetHLModule();
- }
- void Module::ResetHLModule() {
- if (HasHLModule()) {
- delete TheHLModule;
- TheHLModule = nullptr;
- }
- }
- }
|