https://github.com/joaosaffran updated https://github.com/llvm/llvm-project/pull/124967
>From a7a819c8924e5c56d08ce7dc8ff621392024654d Mon Sep 17 00:00:00 2001 From: joaosaffran <joao.saff...@microsoft.com> Date: Thu, 30 Jan 2025 00:42:27 +0000 Subject: [PATCH 1/4] formating --- llvm/include/llvm/BinaryFormat/DXContainer.h | 7 +++ .../llvm/MC/DXContainerRootSignature.h | 3 +- llvm/lib/MC/DXContainerRootSignature.cpp | 4 ++ .../lib/Target/DirectX/DXContainerGlobals.cpp | 22 ++++++++ llvm/lib/Target/DirectX/DXILRootSignature.cpp | 51 +++++++++++++++++-- llvm/lib/Target/DirectX/DXILRootSignature.h | 38 ++++++++++++++ .../ContainerData/RootSignature-Flags.ll | 17 +++++-- 7 files changed, 132 insertions(+), 10 deletions(-) diff --git a/llvm/include/llvm/BinaryFormat/DXContainer.h b/llvm/include/llvm/BinaryFormat/DXContainer.h index cb57309ae183fc..50c30e509fbae6 100644 --- a/llvm/include/llvm/BinaryFormat/DXContainer.h +++ b/llvm/include/llvm/BinaryFormat/DXContainer.h @@ -86,6 +86,13 @@ struct RootConstants { }; struct RootParameter { + RootParameter() = default; + RootParameter(RootConstants RootConstant, ShaderVisibilityFlag Visibility) { + ParameterType = RootParameterType::Constants32Bit; + Constants = RootConstant; + ShaderVisibility = Visibility; + } + RootParameterType ParameterType; union { RootConstants Constants; diff --git a/llvm/include/llvm/MC/DXContainerRootSignature.h b/llvm/include/llvm/MC/DXContainerRootSignature.h index 63a5699a978b79..f0228c5c6f1bc4 100644 --- a/llvm/include/llvm/MC/DXContainerRootSignature.h +++ b/llvm/include/llvm/MC/DXContainerRootSignature.h @@ -8,7 +8,6 @@ #include "llvm/ADT/SmallVector.h" #include "llvm/BinaryFormat/DXContainer.h" -#include <cstdint> #include <limits> namespace llvm { @@ -23,6 +22,8 @@ struct RootSignatureHeader { void swapBytes(); void write(raw_ostream &OS); + + void pushPart(dxbc::RootParameter Part); }; } // namespace mcdxbc } // namespace llvm diff --git a/llvm/lib/MC/DXContainerRootSignature.cpp b/llvm/lib/MC/DXContainerRootSignature.cpp index 5893752a767beb..c0080194ca7f5d 100644 --- a/llvm/lib/MC/DXContainerRootSignature.cpp +++ b/llvm/lib/MC/DXContainerRootSignature.cpp @@ -31,3 +31,7 @@ void RootSignatureHeader::write(raw_ostream &OS) { OS.write(reinterpret_cast<const char *>(&Param), BindingSize); } } + +void RootSignatureHeader::pushPart(dxbc::RootParameter Param){ + Parameters.push_back(Param); +} diff --git a/llvm/lib/Target/DirectX/DXContainerGlobals.cpp b/llvm/lib/Target/DirectX/DXContainerGlobals.cpp index 36e7cedbdaee0c..47ac64fda7f6ef 100644 --- a/llvm/lib/Target/DirectX/DXContainerGlobals.cpp +++ b/llvm/lib/Target/DirectX/DXContainerGlobals.cpp @@ -72,6 +72,23 @@ class DXContainerGlobals : public llvm::ModulePass { } // namespace +static dxbc::RootParameter constructHeaderPart(const RootSignaturePart &Part) { + + dxbc::ShaderVisibilityFlag Visibility = static_cast<dxbc::ShaderVisibilityFlag>(Part.Visibility); + + switch(Part.Type){ + + case PartType::Constants:{ + + return dxbc::RootParameter(dxbc::RootConstants { + Part.Constants.ShaderRegistry, + Part.Constants.RegistrySpace, + Part.Constants.Number32BitValues + }, Visibility); + } break; + } +} + bool DXContainerGlobals::runOnModule(Module &M) { llvm::SmallVector<GlobalValue *> Globals; Globals.push_back(getFeatureFlags(M)); @@ -163,6 +180,11 @@ void DXContainerGlobals::addRootSignature(Module &M, RootSignatureHeader RSH; RSH.Flags = MRS->Flags; RSH.Version = MRS->Version; + + for(const auto &Part : MRS->Parts){ + RSH.pushPart(constructHeaderPart(Part)); + } + RSH.write(OS); Constant *Constant = diff --git a/llvm/lib/Target/DirectX/DXILRootSignature.cpp b/llvm/lib/Target/DirectX/DXILRootSignature.cpp index 52c7ad8e249379..8e01216c8b42f6 100644 --- a/llvm/lib/Target/DirectX/DXILRootSignature.cpp +++ b/llvm/lib/Target/DirectX/DXILRootSignature.cpp @@ -26,13 +26,23 @@ static bool reportError(Twine Message) { return true; } + +static bool isValidShaderVisibility(uint32_t V) { + return V < static_cast<uint32_t>(ShaderVisibility::MAX_VALUE); +} + + +static uint64_t extractInt(MDNode *Node, unsigned int I) { + assert(I > 0 && I < Node->getNumOperands() && "Invalid operand Index"); + return mdconst::extract<ConstantInt>(Node->getOperand(I))->getZExtValue(); +} + + static bool parseRootFlags(ModuleRootSignature *MRS, MDNode *RootFlagNode) { if (RootFlagNode->getNumOperands() != 2) return reportError("Invalid format for RootFlag Element"); - - auto *Flag = mdconst::extract<ConstantInt>(RootFlagNode->getOperand(1)); - uint32_t Value = Flag->getZExtValue(); + uint64_t Value = extractInt(RootFlagNode, 1); // Root Element validation, as specified: // https://github.com/llvm/wg-hlsl/blob/main/proposals/0002-root-signature-in-clang.md#validations-during-dxil-generation @@ -43,6 +53,36 @@ static bool parseRootFlags(ModuleRootSignature *MRS, MDNode *RootFlagNode) { return false; } +static bool parseRootConstants(ModuleRootSignature *MRS, MDNode *RootFlagNode) { + assert(RootFlagNode->getNumOperands() == 5 && + "Invalid format for RootFlag Element"); + + uint32_t MaybeShaderVisibility = extractInt(RootFlagNode, 1); + assert(isValidShaderVisibility(MaybeShaderVisibility) && "Invalid shader visibility value"); + + ShaderVisibility Visibility = static_cast<ShaderVisibility>(MaybeShaderVisibility); + + uint32_t ShaderRegistry = extractInt(RootFlagNode, 2); + uint32_t RegisterSpace = extractInt(RootFlagNode, 3); + uint32_t Num32BitsValue = extractInt(RootFlagNode, 4); + + RootConstants Constant { + ShaderRegistry, + RegisterSpace, + Num32BitsValue + }; + + RootSignaturePart Part { + PartType::Constants, + {Constant}, + Visibility + }; + + MRS->pushPart(Part); + + return false; +} + static bool parseRootSignatureElement(ModuleRootSignature *MRS, MDNode *Element) { MDString *ElementText = cast<MDString>(Element->getOperand(0)); @@ -68,7 +108,10 @@ static bool parseRootSignatureElement(ModuleRootSignature *MRS, break; } - case RootSignatureElementKind::RootConstants: + case RootSignatureElementKind::RootConstants:{ + return parseRootConstants(MRS, Element); + break; + } case RootSignatureElementKind::RootDescriptor: case RootSignatureElementKind::DescriptorTable: case RootSignatureElementKind::StaticSampler: diff --git a/llvm/lib/Target/DirectX/DXILRootSignature.h b/llvm/lib/Target/DirectX/DXILRootSignature.h index 0439deea6451a6..a39f6112cfcdb7 100644 --- a/llvm/lib/Target/DirectX/DXILRootSignature.h +++ b/llvm/lib/Target/DirectX/DXILRootSignature.h @@ -12,9 +12,11 @@ /// //===----------------------------------------------------------------------===// +#include "llvm/ADT/SmallVector.h" #include "llvm/IR/Metadata.h" #include "llvm/IR/PassManager.h" #include "llvm/Pass.h" +#include <cstdint> #include <optional> namespace llvm { @@ -29,15 +31,51 @@ enum class RootSignatureElementKind { StaticSampler = 5 }; +enum class PartType { + Constants = 0 +}; + +enum class ShaderVisibility : uint32_t { + SHADER_VISIBILITY_ALL = 0, + SHADER_VISIBILITY_VERTEX = 1, + SHADER_VISIBILITY_HULL = 2, + SHADER_VISIBILITY_DOMAIN = 3, + SHADER_VISIBILITY_GEOMETRY =4 , + SHADER_VISIBILITY_PIXEL = 5, + SHADER_VISIBILITY_AMPLIFICATION = 6, + SHADER_VISIBILITY_MESH = 7, + // not a flag + MAX_VALUE = 8 +}; + +struct RootConstants { + uint32_t ShaderRegistry; + uint32_t RegistrySpace; + uint32_t Number32BitValues; +}; + +struct RootSignaturePart { + PartType Type; + union { + RootConstants Constants; + }; + ShaderVisibility Visibility; +}; + struct ModuleRootSignature { uint32_t Version; uint32_t Flags; + SmallVector<RootSignaturePart> Parts; ModuleRootSignature() = default; bool parse(int32_t Version, NamedMDNode *Root); static ModuleRootSignature analyzeModule(Module &M); + + void pushPart(RootSignaturePart Part) { + Parts.push_back(Part); + } }; class RootSignatureAnalysis : public AnalysisInfoMixin<RootSignatureAnalysis> { diff --git a/llvm/test/CodeGen/DirectX/ContainerData/RootSignature-Flags.ll b/llvm/test/CodeGen/DirectX/ContainerData/RootSignature-Flags.ll index 4914a409e65286..fdfddf83b1162a 100644 --- a/llvm/test/CodeGen/DirectX/ContainerData/RootSignature-Flags.ll +++ b/llvm/test/CodeGen/DirectX/ContainerData/RootSignature-Flags.ll @@ -3,7 +3,7 @@ target triple = "dxil-unknown-shadermodel6.0-compute" -; CHECK: @dx.rts0 = private constant [16 x i8] c"{{.*}}", section "RTS0", align 4 +; CHECK: @dx.rts0 = private constant [40 x i8] c"{{.*}}", section "RTS0", align 4 define void @main() #0 { @@ -16,15 +16,22 @@ attributes #0 = { "hlsl.numthreads"="1,1,1" "hlsl.shader"="compute" } !dx.rootsignatures = !{!2} ; list of function/root signature pairs !2 = !{ ptr @main, !3 } ; function, root signature -!3 = !{ !4 } ; list of root signature elements +!3 = !{ !4, !5 } ; list of root signature elements !4 = !{ !"RootFlags", i32 1 } ; 1 = allow_input_assembler_input_layout +!5 = !{ !"RootConstants", i32 0, i32 1, i32 2, i32 3 } ; DXC: - Name: RTS0 -; DXC-NEXT: Size: 16 +; DXC-NEXT: Size: 40 ; DXC-NEXT: RootSignature: ; DXC-NEXT: Size: 64 ; DXC-NEXT: Version: 1 -; DXC-NEXT: NumParameters: 0 -; DXC-NEXT: Parameters: [] +; DXC-NEXT: NumParameters: 1 +; DXC-NEXT: Parameters: +; DXC-NEXT: - Type: Constants32Bit +; DXC-NEXT: ShaderVisibility: All +; DXC-NEXT: Constants: +; DXC-NEXT: Num32BitValues: 3 +; DXC-NEXT: ShaderRegister: 1 +; DXC-NEXT: RegisterSpace: 2 ; DXC-NEXT: AllowInputAssemblerInputLayout: true >From dc35c5509377d5ceaae1cea34912d08964bbe80f Mon Sep 17 00:00:00 2001 From: joaosaffran <joao.saff...@microsoft.com> Date: Wed, 29 Jan 2025 00:01:43 +0000 Subject: [PATCH 2/4] adding root constants support to DXIL --- llvm/include/llvm/BinaryFormat/DXContainer.h | 2 +- llvm/lib/MC/DXContainerRootSignature.cpp | 2 +- .../lib/Target/DirectX/DXContainerGlobals.cpp | 21 ++++----- llvm/lib/Target/DirectX/DXILRootSignature.cpp | 43 ++++++++----------- llvm/lib/Target/DirectX/DXILRootSignature.h | 20 ++++----- 5 files changed, 38 insertions(+), 50 deletions(-) diff --git a/llvm/include/llvm/BinaryFormat/DXContainer.h b/llvm/include/llvm/BinaryFormat/DXContainer.h index 50c30e509fbae6..25bada1fc0ef0f 100644 --- a/llvm/include/llvm/BinaryFormat/DXContainer.h +++ b/llvm/include/llvm/BinaryFormat/DXContainer.h @@ -92,7 +92,7 @@ struct RootParameter { Constants = RootConstant; ShaderVisibility = Visibility; } - + RootParameterType ParameterType; union { RootConstants Constants; diff --git a/llvm/lib/MC/DXContainerRootSignature.cpp b/llvm/lib/MC/DXContainerRootSignature.cpp index c0080194ca7f5d..60df2d3934faa4 100644 --- a/llvm/lib/MC/DXContainerRootSignature.cpp +++ b/llvm/lib/MC/DXContainerRootSignature.cpp @@ -32,6 +32,6 @@ void RootSignatureHeader::write(raw_ostream &OS) { } } -void RootSignatureHeader::pushPart(dxbc::RootParameter Param){ +void RootSignatureHeader::pushPart(dxbc::RootParameter Param) { Parameters.push_back(Param); } diff --git a/llvm/lib/Target/DirectX/DXContainerGlobals.cpp b/llvm/lib/Target/DirectX/DXContainerGlobals.cpp index 47ac64fda7f6ef..3481e8a2796537 100644 --- a/llvm/lib/Target/DirectX/DXContainerGlobals.cpp +++ b/llvm/lib/Target/DirectX/DXContainerGlobals.cpp @@ -73,18 +73,19 @@ class DXContainerGlobals : public llvm::ModulePass { } // namespace static dxbc::RootParameter constructHeaderPart(const RootSignaturePart &Part) { - - dxbc::ShaderVisibilityFlag Visibility = static_cast<dxbc::ShaderVisibilityFlag>(Part.Visibility); - switch(Part.Type){ + dxbc::ShaderVisibilityFlag Visibility = + static_cast<dxbc::ShaderVisibilityFlag>(Part.Visibility); - case PartType::Constants:{ + switch (Part.Type) { - return dxbc::RootParameter(dxbc::RootConstants { - Part.Constants.ShaderRegistry, - Part.Constants.RegistrySpace, - Part.Constants.Number32BitValues - }, Visibility); + case PartType::Constants: { + + return dxbc::RootParameter( + dxbc::RootConstants{Part.Constants.ShaderRegistry, + Part.Constants.RegistrySpace, + Part.Constants.Number32BitValues}, + Visibility); } break; } } @@ -181,7 +182,7 @@ void DXContainerGlobals::addRootSignature(Module &M, RSH.Flags = MRS->Flags; RSH.Version = MRS->Version; - for(const auto &Part : MRS->Parts){ + for (const auto &Part : MRS->Parts) { RSH.pushPart(constructHeaderPart(Part)); } diff --git a/llvm/lib/Target/DirectX/DXILRootSignature.cpp b/llvm/lib/Target/DirectX/DXILRootSignature.cpp index 8e01216c8b42f6..6260ff2c24d05a 100644 --- a/llvm/lib/Target/DirectX/DXILRootSignature.cpp +++ b/llvm/lib/Target/DirectX/DXILRootSignature.cpp @@ -26,18 +26,15 @@ static bool reportError(Twine Message) { return true; } - static bool isValidShaderVisibility(uint32_t V) { - return V < static_cast<uint32_t>(ShaderVisibility::MAX_VALUE); + return V < static_cast<uint32_t>(ShaderVisibility::MAX_VALUE); } - static uint64_t extractInt(MDNode *Node, unsigned int I) { assert(I > 0 && I < Node->getNumOperands() && "Invalid operand Index"); return mdconst::extract<ConstantInt>(Node->getOperand(I))->getZExtValue(); } - static bool parseRootFlags(ModuleRootSignature *MRS, MDNode *RootFlagNode) { if (RootFlagNode->getNumOperands() != 2) @@ -56,27 +53,21 @@ static bool parseRootFlags(ModuleRootSignature *MRS, MDNode *RootFlagNode) { static bool parseRootConstants(ModuleRootSignature *MRS, MDNode *RootFlagNode) { assert(RootFlagNode->getNumOperands() == 5 && "Invalid format for RootFlag Element"); - - uint32_t MaybeShaderVisibility = extractInt(RootFlagNode, 1); - assert(isValidShaderVisibility(MaybeShaderVisibility) && "Invalid shader visibility value"); - - ShaderVisibility Visibility = static_cast<ShaderVisibility>(MaybeShaderVisibility); - - uint32_t ShaderRegistry = extractInt(RootFlagNode, 2); - uint32_t RegisterSpace = extractInt(RootFlagNode, 3); - uint32_t Num32BitsValue = extractInt(RootFlagNode, 4); - - RootConstants Constant { - ShaderRegistry, - RegisterSpace, - Num32BitsValue - }; - - RootSignaturePart Part { - PartType::Constants, - {Constant}, - Visibility - }; + + uint32_t MaybeShaderVisibility = extractInt(RootFlagNode, 1); + assert(isValidShaderVisibility(MaybeShaderVisibility) && + "Invalid shader visibility value"); + + ShaderVisibility Visibility = + static_cast<ShaderVisibility>(MaybeShaderVisibility); + + uint32_t ShaderRegistry = extractInt(RootFlagNode, 2); + uint32_t RegisterSpace = extractInt(RootFlagNode, 3); + uint32_t Num32BitsValue = extractInt(RootFlagNode, 4); + + RootConstants Constant{ShaderRegistry, RegisterSpace, Num32BitsValue}; + + RootSignaturePart Part{PartType::Constants, {Constant}, Visibility}; MRS->pushPart(Part); @@ -108,7 +99,7 @@ static bool parseRootSignatureElement(ModuleRootSignature *MRS, break; } - case RootSignatureElementKind::RootConstants:{ + case RootSignatureElementKind::RootConstants: { return parseRootConstants(MRS, Element); break; } diff --git a/llvm/lib/Target/DirectX/DXILRootSignature.h b/llvm/lib/Target/DirectX/DXILRootSignature.h index a39f6112cfcdb7..1aa4848ee5ffe4 100644 --- a/llvm/lib/Target/DirectX/DXILRootSignature.h +++ b/llvm/lib/Target/DirectX/DXILRootSignature.h @@ -31,16 +31,14 @@ enum class RootSignatureElementKind { StaticSampler = 5 }; -enum class PartType { - Constants = 0 -}; +enum class PartType { Constants = 0 }; enum class ShaderVisibility : uint32_t { SHADER_VISIBILITY_ALL = 0, SHADER_VISIBILITY_VERTEX = 1, SHADER_VISIBILITY_HULL = 2, SHADER_VISIBILITY_DOMAIN = 3, - SHADER_VISIBILITY_GEOMETRY =4 , + SHADER_VISIBILITY_GEOMETRY = 4, SHADER_VISIBILITY_PIXEL = 5, SHADER_VISIBILITY_AMPLIFICATION = 6, SHADER_VISIBILITY_MESH = 7, @@ -55,11 +53,11 @@ struct RootConstants { }; struct RootSignaturePart { - PartType Type; - union { - RootConstants Constants; - }; - ShaderVisibility Visibility; + PartType Type; + union { + RootConstants Constants; + }; + ShaderVisibility Visibility; }; struct ModuleRootSignature { @@ -73,9 +71,7 @@ struct ModuleRootSignature { static ModuleRootSignature analyzeModule(Module &M); - void pushPart(RootSignaturePart Part) { - Parts.push_back(Part); - } + void pushPart(RootSignaturePart Part) { Parts.push_back(Part); } }; class RootSignatureAnalysis : public AnalysisInfoMixin<RootSignatureAnalysis> { >From 1a27dfc074cc4b175db0a0ef08a1c73e0003460c Mon Sep 17 00:00:00 2001 From: joaosaffran <joao.saff...@microsoft.com> Date: Wed, 29 Jan 2025 18:19:20 +0000 Subject: [PATCH 3/4] clean up --- llvm/lib/Target/DirectX/DXILRootSignature.h | 1 - 1 file changed, 1 deletion(-) diff --git a/llvm/lib/Target/DirectX/DXILRootSignature.h b/llvm/lib/Target/DirectX/DXILRootSignature.h index 1aa4848ee5ffe4..ab71a3fef299ab 100644 --- a/llvm/lib/Target/DirectX/DXILRootSignature.h +++ b/llvm/lib/Target/DirectX/DXILRootSignature.h @@ -16,7 +16,6 @@ #include "llvm/IR/Metadata.h" #include "llvm/IR/PassManager.h" #include "llvm/Pass.h" -#include <cstdint> #include <optional> namespace llvm { >From 2dba1bd4a7ae64b9c1ac40a4cfffab94f05b43b2 Mon Sep 17 00:00:00 2001 From: joaosaffran <joao.saff...@microsoft.com> Date: Thu, 30 Jan 2025 18:33:23 +0000 Subject: [PATCH 4/4] removing version --- llvm/include/llvm/MC/DXContainerRootSignature.h | 1 - llvm/include/llvm/Object/DXContainer.h | 3 --- llvm/include/llvm/ObjectYAML/DXContainerYAML.h | 1 - llvm/lib/MC/DXContainerRootSignature.cpp | 1 - llvm/lib/Object/DXContainer.cpp | 3 --- llvm/lib/ObjectYAML/DXContainerEmitter.cpp | 1 - llvm/lib/ObjectYAML/DXContainerYAML.cpp | 4 +--- llvm/lib/Target/DirectX/DXContainerGlobals.cpp | 1 - llvm/lib/Target/DirectX/DXILRootSignature.cpp | 5 ++--- llvm/lib/Target/DirectX/DXILRootSignature.h | 3 +-- .../CodeGen/DirectX/ContainerData/RootSignature-Flags.ll | 5 ++--- llvm/test/ObjectYAML/DXContainer/RootSignature-Flags.yaml | 8 +++----- 12 files changed, 9 insertions(+), 27 deletions(-) diff --git a/llvm/include/llvm/MC/DXContainerRootSignature.h b/llvm/include/llvm/MC/DXContainerRootSignature.h index f0228c5c6f1bc4..e797953f2c3067 100644 --- a/llvm/include/llvm/MC/DXContainerRootSignature.h +++ b/llvm/include/llvm/MC/DXContainerRootSignature.h @@ -16,7 +16,6 @@ class raw_ostream; namespace mcdxbc { struct RootSignatureHeader { - uint32_t Version; uint32_t Flags; SmallVector<dxbc::RootParameter> Parameters; diff --git a/llvm/include/llvm/Object/DXContainer.h b/llvm/include/llvm/Object/DXContainer.h index be13ee3ea8680a..e2d9f7266fcb49 100644 --- a/llvm/include/llvm/Object/DXContainer.h +++ b/llvm/include/llvm/Object/DXContainer.h @@ -124,7 +124,6 @@ class RootSignature { private: StringRef Data; uint32_t Size; - uint32_t Version; uint32_t Flags; uint32_t NParameters; ParametersArray Parameters; @@ -136,8 +135,6 @@ class RootSignature { uint32_t getSize() const { return Size; } - uint32_t getVersion() const { return Version; } - uint32_t getFlags() const { return Flags; } uint32_t getNParameters() const { return NParameters; } diff --git a/llvm/include/llvm/ObjectYAML/DXContainerYAML.h b/llvm/include/llvm/ObjectYAML/DXContainerYAML.h index b1366c8de3dc49..a368d5bf3634af 100644 --- a/llvm/include/llvm/ObjectYAML/DXContainerYAML.h +++ b/llvm/include/llvm/ObjectYAML/DXContainerYAML.h @@ -80,7 +80,6 @@ struct RootSignatureDesc { uint32_t getEncodedFlags(); uint32_t Size; - uint32_t Version; uint32_t NumParameters; SmallVector<dxbc::RootParameter> Parameters; diff --git a/llvm/lib/MC/DXContainerRootSignature.cpp b/llvm/lib/MC/DXContainerRootSignature.cpp index 60df2d3934faa4..b4370ef768a945 100644 --- a/llvm/lib/MC/DXContainerRootSignature.cpp +++ b/llvm/lib/MC/DXContainerRootSignature.cpp @@ -18,7 +18,6 @@ void RootSignatureHeader::write(raw_ostream &OS) { uint32_t SizeInfo = sizeof(RootSignatureHeader); uint32_t ParamsSize = Parameters.size(); support::endian::write(OS, SizeInfo, llvm::endianness::little); - support::endian::write(OS, Version, llvm::endianness::little); support::endian::write(OS, Flags, llvm::endianness::little); support::endian::write(OS, ParamsSize, llvm::endianness::little); diff --git a/llvm/lib/Object/DXContainer.cpp b/llvm/lib/Object/DXContainer.cpp index 48bbf22a12ff9d..627433e38f51c0 100644 --- a/llvm/lib/Object/DXContainer.cpp +++ b/llvm/lib/Object/DXContainer.cpp @@ -248,9 +248,6 @@ Error DirectX::RootSignature::parse() { Size = support::endian::read<uint32_t, llvm::endianness::little>(Current); Current += sizeof(uint32_t); - Version = support::endian::read<uint32_t, llvm::endianness::little>(Current); - Current += sizeof(uint32_t); - Flags = support::endian::read<uint32_t, llvm::endianness::little>(Current); Current += sizeof(uint32_t); diff --git a/llvm/lib/ObjectYAML/DXContainerEmitter.cpp b/llvm/lib/ObjectYAML/DXContainerEmitter.cpp index 9aeda0f12ead87..ecf3714d29d613 100644 --- a/llvm/lib/ObjectYAML/DXContainerEmitter.cpp +++ b/llvm/lib/ObjectYAML/DXContainerEmitter.cpp @@ -267,7 +267,6 @@ void DXContainerWriter::writeParts(raw_ostream &OS) { continue; mcdxbc::RootSignatureHeader Header; - Header.Version = P.RootSignature->Version; Header.Flags = P.RootSignature->getEncodedFlags(); Header.Parameters = P.RootSignature->Parameters; diff --git a/llvm/lib/ObjectYAML/DXContainerYAML.cpp b/llvm/lib/ObjectYAML/DXContainerYAML.cpp index 184bf32279141a..3dbe174ac4cc88 100644 --- a/llvm/lib/ObjectYAML/DXContainerYAML.cpp +++ b/llvm/lib/ObjectYAML/DXContainerYAML.cpp @@ -32,8 +32,7 @@ DXContainerYAML::ShaderFeatureFlags::ShaderFeatureFlags(uint64_t FlagData) { DXContainerYAML::RootSignatureDesc::RootSignatureDesc( const object::DirectX::RootSignature &Data) - : Size(Data.getSize()), Version(Data.getVersion()), - NumParameters(Data.getNParameters()) { + : Size(Data.getSize()), NumParameters(Data.getNParameters()) { uint32_t Flags = Data.getFlags(); for (const auto &Param : Data.getParameters()) @@ -215,7 +214,6 @@ void MappingTraits<DXContainerYAML::Signature>::mapping( void MappingTraits<DXContainerYAML::RootSignatureDesc>::mapping( IO &IO, DXContainerYAML::RootSignatureDesc &S) { IO.mapRequired("Size", S.Size); - IO.mapRequired("Version", S.Version); IO.mapRequired("NumParameters", S.NumParameters); IO.mapRequired("Parameters", S.Parameters); #define ROOT_ELEMENT_FLAG(Num, Val, Str) IO.mapOptional(#Val, S.Val, false); diff --git a/llvm/lib/Target/DirectX/DXContainerGlobals.cpp b/llvm/lib/Target/DirectX/DXContainerGlobals.cpp index 3481e8a2796537..7c229ace5e502c 100644 --- a/llvm/lib/Target/DirectX/DXContainerGlobals.cpp +++ b/llvm/lib/Target/DirectX/DXContainerGlobals.cpp @@ -180,7 +180,6 @@ void DXContainerGlobals::addRootSignature(Module &M, RootSignatureHeader RSH; RSH.Flags = MRS->Flags; - RSH.Version = MRS->Version; for (const auto &Part : MRS->Parts) { RSH.pushPart(constructHeaderPart(Part)); diff --git a/llvm/lib/Target/DirectX/DXILRootSignature.cpp b/llvm/lib/Target/DirectX/DXILRootSignature.cpp index 6260ff2c24d05a..b4996c9a47d688 100644 --- a/llvm/lib/Target/DirectX/DXILRootSignature.cpp +++ b/llvm/lib/Target/DirectX/DXILRootSignature.cpp @@ -114,8 +114,7 @@ static bool parseRootSignatureElement(ModuleRootSignature *MRS, return true; } -bool ModuleRootSignature::parse(int32_t Version, NamedMDNode *Root) { - this->Version = Version; +bool ModuleRootSignature::parse(NamedMDNode *Root) { bool HasError = false; /** Root Signature are specified as following in the metadata: @@ -161,7 +160,7 @@ ModuleRootSignature ModuleRootSignature::analyzeModule(Module &M) { NamedMDNode *RootSignatureNode = M.getNamedMetadata("dx.rootsignatures"); if (RootSignatureNode) { - if (MRS.parse(1, RootSignatureNode)) + if (MRS.parse(RootSignatureNode)) llvm_unreachable("Invalid Root Signature Metadata."); } diff --git a/llvm/lib/Target/DirectX/DXILRootSignature.h b/llvm/lib/Target/DirectX/DXILRootSignature.h index ab71a3fef299ab..33e4f3b033f315 100644 --- a/llvm/lib/Target/DirectX/DXILRootSignature.h +++ b/llvm/lib/Target/DirectX/DXILRootSignature.h @@ -60,13 +60,12 @@ struct RootSignaturePart { }; struct ModuleRootSignature { - uint32_t Version; uint32_t Flags; SmallVector<RootSignaturePart> Parts; ModuleRootSignature() = default; - bool parse(int32_t Version, NamedMDNode *Root); + bool parse(NamedMDNode *Root); static ModuleRootSignature analyzeModule(Module &M); diff --git a/llvm/test/CodeGen/DirectX/ContainerData/RootSignature-Flags.ll b/llvm/test/CodeGen/DirectX/ContainerData/RootSignature-Flags.ll index fdfddf83b1162a..a1348fa439227d 100644 --- a/llvm/test/CodeGen/DirectX/ContainerData/RootSignature-Flags.ll +++ b/llvm/test/CodeGen/DirectX/ContainerData/RootSignature-Flags.ll @@ -3,7 +3,7 @@ target triple = "dxil-unknown-shadermodel6.0-compute" -; CHECK: @dx.rts0 = private constant [40 x i8] c"{{.*}}", section "RTS0", align 4 +; CHECK: @dx.rts0 = private constant [36 x i8] c"{{.*}}", section "RTS0", align 4 define void @main() #0 { @@ -22,10 +22,9 @@ attributes #0 = { "hlsl.numthreads"="1,1,1" "hlsl.shader"="compute" } ; DXC: - Name: RTS0 -; DXC-NEXT: Size: 40 +; DXC-NEXT: Size: 36 ; DXC-NEXT: RootSignature: ; DXC-NEXT: Size: 64 -; DXC-NEXT: Version: 1 ; DXC-NEXT: NumParameters: 1 ; DXC-NEXT: Parameters: ; DXC-NEXT: - Type: Constants32Bit diff --git a/llvm/test/ObjectYAML/DXContainer/RootSignature-Flags.yaml b/llvm/test/ObjectYAML/DXContainer/RootSignature-Flags.yaml index b719768e75058c..2da4ef8db96825 100644 --- a/llvm/test/ObjectYAML/DXContainer/RootSignature-Flags.yaml +++ b/llvm/test/ObjectYAML/DXContainer/RootSignature-Flags.yaml @@ -10,10 +10,9 @@ Header: PartOffsets: [ 60 ] Parts: - Name: RTS0 - Size: 44 + Size: 40 RootSignature: - Size: 44 - Version: 1 + Size: 64 NumParameters: 1 Parameters: - Type: Constants32Bit @@ -25,10 +24,9 @@ Parts: AllowInputAssemblerInputLayout: true #CHECK: - Name: RTS0 -#CHECK-NEXT: Size: 44 +#CHECK-NEXT: Size: 40 #CHECK-NEXT: RootSignature: #CHECK-NEXT: Size: 64 -#CHECK-NEXT: Version: 1 #CHECK-NEXT: NumParameters: 1 #CHECK-NEXT: Parameters: #CHECK-NEXT: - Type: Constants32Bit _______________________________________________ llvm-branch-commits mailing list llvm-branch-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits