Skip to content

Commit 3a10a5c

Browse files
hekotahekota
committed
[DirectX] Update resource type names in DXIL metadata to include element type
1 parent 69c30f7 commit 3a10a5c

File tree

7 files changed

+152
-76
lines changed

7 files changed

+152
-76
lines changed

‎llvm/lib/Analysis/DXILResource.cpp‎

Lines changed: 97 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ static StringRef getResourceKindName(ResourceKind RK) {
6464
case ResourceKind::TextureCubeArray:
6565
return "TextureCubeArray";
6666
case ResourceKind::TypedBuffer:
67-
return "TypedBuffer";
67+
return "Buffer";
6868
case ResourceKind::RawBuffer:
6969
return "RawBuffer";
7070
case ResourceKind::StructuredBuffer:
@@ -132,6 +132,44 @@ static StringRef getElementTypeName(ElementType ET) {
132132
llvm_unreachable("Unhandled ElementType");
133133
}
134134

135+
static StringRef getElementTypeNameForTemplate(ElementType ET) {
136+
switch (ET) {
137+
case ElementType::I1:
138+
return "bool";
139+
case ElementType::I16:
140+
return "int16_t";
141+
case ElementType::U16:
142+
return "uint16_t";
143+
case ElementType::I32:
144+
return "int32_t";
145+
case ElementType::U32:
146+
return "uint32_t";
147+
case ElementType::I64:
148+
return "int64_t";
149+
case ElementType::U64:
150+
return "uint32_t";
151+
case ElementType::F16:
152+
case ElementType::SNormF16:
153+
case ElementType::UNormF16:
154+
return "half";
155+
case ElementType::F32:
156+
case ElementType::SNormF32:
157+
case ElementType::UNormF32:
158+
return "float";
159+
case ElementType::F64:
160+
case ElementType::SNormF64:
161+
case ElementType::UNormF64:
162+
return "double";
163+
case ElementType::PackedS8x32:
164+
return "int8_t4_packed";
165+
case ElementType::PackedU8x32:
166+
return "uint8_t4_packed";
167+
case ElementType::Invalid:
168+
return "<invalid>";
169+
}
170+
llvm_unreachable("Unhandled ElementType");
171+
}
172+
135173
static StringRef getSamplerTypeName(SamplerType ST) {
136174
switch (ST) {
137175
case SamplerType::Default:
@@ -219,11 +257,46 @@ ResourceTypeInfo::ResourceTypeInfo(TargetExtType *HandleTy,
219257
}
220258

221259
static void formatTypeName(SmallString<64> &Dest, StringRef Name,
222-
bool isWriteable, bool isROV) {
223-
Dest = isWriteable ? (isROV ? "RasterizerOrdered" : "RW") : "";
260+
bool IsWriteable, bool IsROV) {
261+
Dest = IsWriteable ? (IsROV ? "RasterizerOrdered" : "RW") : "";
224262
Dest += Name;
225263
}
226264

265+
static void formatTypeName(SmallString<64> &Dest, StringRef Name,
266+
bool IsWriteable, bool IsROV, Type *ContainedType,
267+
bool IsSigned) {
268+
raw_svector_ostream DestStream(Dest);
269+
if (IsWriteable)
270+
DestStream << (IsROV ? "RasterizerOrdered" : "RW");
271+
DestStream << Name;
272+
273+
if (ContainedType) {
274+
StringRef ElementName;
275+
ElementType ET = toDXILElementType(ContainedType, IsSigned);
276+
if (ET != ElementType::Invalid) {
277+
ElementName = getElementTypeNameForTemplate(ET);
278+
} else if (StructType *ST = dyn_cast<StructType>(ContainedType)) {
279+
if (!ST->hasName())
280+
return;
281+
ElementName = ST->getStructName();
282+
} else {
283+
llvm_unreachable("invalid element type for raw buffer");
284+
}
285+
286+
if (const FixedVectorType *VTy = dyn_cast<FixedVectorType>(ContainedType))
287+
DestStream << "<" << ElementName << VTy->getNumElements() << ">";
288+
else
289+
DestStream << "<" << ElementName << ">";
290+
}
291+
}
292+
293+
static StructType *getOrCreateElementStruct(Type *ElemType, StringRef Name) {
294+
StructType *Ty = StructType::getTypeByName(ElemType->getContext(), Name);
295+
if (Ty && Ty->getNumElements() == 1 && Ty->getElementType(0) == ElemType)
296+
return Ty;
297+
return StructType::create(ElemType, Name);
298+
}
299+
227300
StructType *ResourceTypeInfo::createElementStruct() {
228301
SmallString<64> TypeName;
229302

@@ -237,51 +310,56 @@ StructType *ResourceTypeInfo::createElementStruct() {
237310
case ResourceKind::TextureCubeArray: {
238311
auto *RTy = cast<TextureExtType>(HandleTy);
239312
formatTypeName(TypeName, getResourceKindName(Kind), RTy->isWriteable(),
240-
RTy->isROV());
241-
return StructType::create(RTy->getResourceType(), TypeName);
313+
RTy->isROV(), RTy->getResourceType(), RTy->isSigned());
314+
return getOrCreateElementStruct(RTy->getResourceType(), TypeName);
242315
}
243316
case ResourceKind::Texture2DMS:
244317
case ResourceKind::Texture2DMSArray: {
245318
auto *RTy = cast<MSTextureExtType>(HandleTy);
246319
formatTypeName(TypeName, getResourceKindName(Kind), RTy->isWriteable(),
247-
/*IsROV=*/false);
248-
return StructType::create(RTy->getResourceType(), TypeName);
320+
/*IsROV=*/false, RTy->getResourceType(), RTy->isSigned());
321+
return getOrCreateElementStruct(RTy->getResourceType(), TypeName);
249322
}
250323
case ResourceKind::TypedBuffer: {
251324
auto *RTy = cast<TypedBufferExtType>(HandleTy);
252325
formatTypeName(TypeName, getResourceKindName(Kind), RTy->isWriteable(),
253-
RTy->isROV());
254-
return StructType::create(RTy->getResourceType(), TypeName);
326+
RTy->isROV(), RTy->getResourceType(), RTy->isSigned());
327+
return getOrCreateElementStruct(RTy->getResourceType(), TypeName);
255328
}
256329
case ResourceKind::RawBuffer: {
257330
auto *RTy = cast<RawBufferExtType>(HandleTy);
258331
formatTypeName(TypeName, "ByteAddressBuffer", RTy->isWriteable(),
259332
RTy->isROV());
260-
return StructType::create(Type::getInt32Ty(HandleTy->getContext()),
261-
TypeName);
333+
return getOrCreateElementStruct(Type::getInt32Ty(HandleTy->getContext()),
334+
TypeName);
262335
}
263336
case ResourceKind::StructuredBuffer: {
264337
auto *RTy = cast<RawBufferExtType>(HandleTy);
338+
Type *Ty = RTy->getResourceType();
265339
formatTypeName(TypeName, "StructuredBuffer", RTy->isWriteable(),
266-
RTy->isROV());
267-
return StructType::create(RTy->getResourceType(), TypeName);
340+
RTy->isROV(), RTy->getResourceType(), true);
341+
return getOrCreateElementStruct(Ty, TypeName);
268342
}
269343
case ResourceKind::FeedbackTexture2D:
270344
case ResourceKind::FeedbackTexture2DArray: {
271345
auto *RTy = cast<FeedbackTextureExtType>(HandleTy);
272346
TypeName = formatv("{0}<{1}>", getResourceKindName(Kind),
273347
llvm::to_underlying(RTy->getFeedbackType()));
274-
return StructType::create(Type::getInt32Ty(HandleTy->getContext()),
275-
TypeName);
348+
return getOrCreateElementStruct(Type::getInt32Ty(HandleTy->getContext()),
349+
TypeName);
350+
}
351+
case ResourceKind::CBuffer: {
352+
auto *RTy = cast<CBufferExtType>(HandleTy);
353+
LayoutExtType *LayoutType = cast<LayoutExtType>(RTy->getResourceType());
354+
StructType *Ty = cast<StructType>(LayoutType->getWrappedType());
355+
return StructType::create(Ty->elements(), getResourceKindName(Kind));
276356
}
277-
case ResourceKind::CBuffer:
278-
return StructType::create(HandleTy->getContext(), "cbuffer");
279357
case ResourceKind::Sampler: {
280358
auto *RTy = cast<SamplerExtType>(HandleTy);
281359
TypeName = formatv("SamplerState<{0}>",
282360
llvm::to_underlying(RTy->getSamplerType()));
283-
return StructType::create(Type::getInt32Ty(HandleTy->getContext()),
284-
TypeName);
361+
return getOrCreateElementStruct(Type::getInt32Ty(HandleTy->getContext()),
362+
TypeName);
285363
}
286364
case ResourceKind::TBuffer:
287365
case ResourceKind::RTAccelerationStructure:

‎llvm/test/Analysis/DXILResource/buffer-frombinding.ll‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ define void @test_typedbuffer() {
5353
; CHECK: Lower Bound: 3
5454
; CHECK: Size: 24
5555
; CHECK: Class: SRV
56-
; CHECK: Kind: TypedBuffer
56+
; CHECK: Kind: Buffer
5757
; CHECK: Element Type: u32
5858
; CHECK: Element Count: 4
5959

@@ -70,7 +70,7 @@ define void @test_typedbuffer() {
7070
; CHECK: Globally Coherent: 0
7171
; CHECK: Counter Direction: Unknown
7272
; CHECK: Class: UAV
73-
; CHECK: Kind: TypedBuffer
73+
; CHECK: Kind: Buffer
7474
; CHECK: IsROV: 0
7575
; CHECK: Element Type: i32
7676
; CHECK: Element Count: 1
@@ -89,7 +89,7 @@ define void @test_typedbuffer() {
8989
; CHECK: Globally Coherent: 0
9090
; CHECK: Counter Direction: Decrement
9191
; CHECK: Class: UAV
92-
; CHECK: Kind: TypedBuffer
92+
; CHECK: Kind: Buffer
9393
; CHECK: IsROV: 0
9494
; CHECK: Element Type: f32
9595
; CHECK: Element Count: 4
@@ -112,7 +112,7 @@ define void @test_typedbuffer() {
112112
; CHECK: Globally Coherent: 0
113113
; CHECK: Counter Direction: Increment
114114
; CHECK: Class: UAV
115-
; CHECK: Kind: TypedBuffer
115+
; CHECK: Kind: Buffer
116116
; CHECK: IsROV: 0
117117
; CHECK: Element Type: f32
118118
; CHECK: Element Count: 4
@@ -132,7 +132,7 @@ define void @test_typedbuffer() {
132132
; CHECK: Globally Coherent: 0
133133
; CHECK: Counter Direction: Invalid
134134
; CHECK: Class: UAV
135-
; CHECK: Kind: TypedBuffer
135+
; CHECK: Kind: Buffer
136136
; CHECK: IsROV: 0
137137
; CHECK: Element Type: f32
138138
; CHECK: Element Count: 4

‎llvm/test/CodeGen/DirectX/Metadata/cbuffer-only.ll‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@ target triple = "dxil-pc-shadermodel6.6-compute"
99
define void @cbuffer_is_only_binding() {
1010
%cbuf = call target("dx.CBuffer", target("dx.Layout", {float}, 4, 0))
1111
@llvm.dx.resource.handlefrombinding(i32 1, i32 8, i32 1, i32 0, i1 false, ptr null)
12-
; CHECK: %cbuffer = type
12+
; CHECK: %CBuffer = type { float }
1313

1414
ret void
1515
}
1616

17-
; CHECK: @[[CB0:.*]] = external constant %cbuffer
17+
; CHECK: @[[CB0:.*]] = external constant %CBuffer
1818

1919
; CHECK: !{i32 0, ptr @[[CB0]], !""

‎llvm/test/CodeGen/DirectX/Metadata/cbuffer_metadata.ll‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,9 @@ define void @test() #0 {
6666

6767
attributes #0 = { noinline nounwind "hlsl.shader"="compute" }
6868

69-
; CHECK: @CB1 = external constant %cbuffer
70-
; CHECK: @CB2 = external constant %cbuffer.0
71-
; CHECK: @MyConstants = external constant %cbuffer.1
69+
; CHECK: @CB1 = external constant %CBuffer
70+
; CHECK: @CB2 = external constant %CBuffer.0
71+
; CHECK: @MyConstants = external constant %CBuffer.1
7272

7373
; CHECK: !dx.resources = !{[[ResList:[!][0-9]+]]}
7474

‎llvm/test/CodeGen/DirectX/Metadata/resource-symbols.ll‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -11,22 +11,22 @@ define void @test() {
1111
; Buffer<float4>
1212
%float4 = call target("dx.TypedBuffer", <4 x float>, 0, 0, 0)
1313
@llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 1, i32 0, i1 false, ptr @A.str)
14-
; CHECK: %TypedBuffer = type { <4 x float> }
14+
; CHECK: %"Buffer<float4>" = type { <4 x float> }
1515

1616
; Buffer<int>
1717
%int = call target("dx.TypedBuffer", i32, 0, 0, 1)
1818
@llvm.dx.resource.handlefrombinding(i32 0, i32 1, i32 1, i32 0, i1 false, ptr null)
19-
; CHECK: %TypedBuffer.0 = type { i32 }
19+
; CHECK: %"Buffer<int32_t>" = type { i32 }
2020

2121
; Buffer<uint3>
2222
%uint3 = call target("dx.TypedBuffer", <3 x i32>, 0, 0, 0)
2323
@llvm.dx.resource.handlefrombinding(i32 0, i32 2, i32 1, i32 0, i1 false, ptr null)
24-
; CHECK: %TypedBuffer.1 = type { <3 x i32> }
24+
; CHECK: %"Buffer<uint32_t3>" = type { <3 x i32> }
2525

2626
; StructuredBuffer<S>
2727
%struct0 = call target("dx.RawBuffer", %struct.S, 0, 0)
2828
@llvm.dx.resource.handlefrombinding(i32 0, i32 10, i32 1, i32 0, i1 true, ptr @SB.str)
29-
; CHECK: %StructuredBuffer = type { %struct.S }
29+
; CHECK: %"StructuredBuffer<struct.S>" = type { %struct.S }
3030

3131
; ByteAddressBuffer
3232
%byteaddr = call target("dx.RawBuffer", i8, 0, 0)
@@ -36,10 +36,10 @@ define void @test() {
3636
ret void
3737
}
3838

39-
; CHECK: @[[T0:.*]] = external constant %TypedBuffer
40-
; CHECK-NEXT: @[[T1:.*]] = external constant %TypedBuffer.0
41-
; CHECK-NEXT: @[[T2:.*]] = external constant %TypedBuffer.1
42-
; CHECK-NEXT: @[[S0:.*]] = external constant %StructuredBuffer
39+
; CHECK: @[[T0:.*]] = external constant %"Buffer<float4>"
40+
; CHECK-NEXT: @[[T1:.*]] = external constant %"Buffer<int32_t>"
41+
; CHECK-NEXT: @[[T2:.*]] = external constant %"Buffer<uint32_t3>"
42+
; CHECK-NEXT: @[[S0:.*]] = external constant %"StructuredBuffer<struct.S>"
4343
; CHECK-NEXT: @[[B0:.*]] = external constant %ByteAddressBuffer
4444

4545
; CHECK: !{i32 0, ptr @[[T0]], !"A"

‎llvm/test/CodeGen/DirectX/Metadata/srv_metadata.ll‎

Lines changed: 16 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -78,25 +78,24 @@ define void @test() #0 {
7878

7979
attributes #0 = { noinline nounwind "hlsl.shader"="compute" }
8080

81-
; CHECK: %TypedBuffer = type { <4 x half> }
82-
; CHECK: %TypedBuffer.0 = type { <2 x float> }
83-
; CHECK: %TypedBuffer.1 = type { double }
84-
; CHECK: %TypedBuffer.2 = type { <4 x i32> }
81+
; CHECK: %"Buffer<half4>" = type { <4 x half> }
82+
; CHECK: %"Buffer<float2>" = type { <2 x float> }
83+
; CHECK: %"Buffer<double>" = type { double }
84+
; CHECK: %"Buffer<int32_t4>" = type { <4 x i32> }
8585
; CHECK: %ByteAddressBuffer = type { i32 }
86-
; CHECK: %StructuredBuffer = type { i16 }
87-
; CHECK: %TypedBuffer.3 = type { i64 }
88-
; CHECK: %TypedBuffer.4 = type { <4 x float> }
89-
; CHECK: %TypedBuffer.5 = type { i64 }
90-
91-
; CHECK: @Zero = external constant %TypedBuffer
92-
; CHECK: @One = external constant %TypedBuffer.0
93-
; CHECK: @Two = external constant %TypedBuffer.1
94-
; CHECK: @Three = external constant %TypedBuffer.2
86+
; CHECK: %"StructuredBuffer<int16_t>" = type { i16 }
87+
; CHECK: %"Buffer<uint32_t>" = type { i64 }
88+
; CHECK: %"Buffer<float4>" = type { <4 x float> }
89+
90+
; CHECK: @Zero = external constant %"Buffer<half4>"
91+
; CHECK: @One = external constant %"Buffer<float2>"
92+
; CHECK: @Two = external constant %"Buffer<double>"
93+
; CHECK: @Three = external constant %"Buffer<int32_t4>"
9594
; CHECK: @Four = external constant %ByteAddressBuffer
96-
; CHECK: @Five = external constant %StructuredBuffer
97-
; CHECK: @Six = external constant %TypedBuffer.3
98-
; CHECK: @Array = external constant %TypedBuffer.4
99-
; CHECK: @Seven = external constant %TypedBuffer.5
95+
; CHECK: @Five = external constant %"StructuredBuffer<int16_t>"
96+
; CHECK: @Six = external constant %"Buffer<uint32_t>"
97+
; CHECK: @Array = external constant %"Buffer<float4>"
98+
; CHECK: @Seven = external constant %"Buffer<uint32_t>"
10099

101100
; CHECK: !dx.resources = !{[[ResList:[!][0-9]+]]}
102101

‎llvm/test/CodeGen/DirectX/Metadata/uav_metadata.ll‎

Lines changed: 21 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ define void @test() #0 {
8585
%Array_42_h = call target("dx.TypedBuffer", <4 x float>, 1, 0, 0)
8686
@llvm.dx.resource.handlefrombinding(i32 3, i32 4, i32 100, i32 42, i1 false, ptr @Array.str)
8787

88-
; Same buffer type as Nine
88+
; Same buffer type as Nine - should have the same type in metadata
8989
; RWBuffer<double> Ten : register(u2);
9090
%Ten_h = call target("dx.TypedBuffer", i64, 1, 0, 0)
9191
@llvm.dx.resource.handlefrombinding(i32 5, i32 22, i32 1, i32 0, i1 false, ptr @Ten.str)
@@ -95,31 +95,30 @@ define void @test() #0 {
9595

9696
attributes #0 = { noinline nounwind "hlsl.shader"="compute" }
9797

98-
; CHECK: %RWTypedBuffer = type { <4 x half> }
99-
; CHECK: %RWTypedBuffer.0 = type { <2 x float> }
100-
; CHECK: %RWTypedBuffer.1 = type { double }
101-
; CHECK: %RWTypedBuffer.2 = type { <4 x i32> }
98+
; CHECK: %"RWBuffer<half4>" = type { <4 x half> }
99+
; CHECK: %"RWBuffer<float2>" = type { <2 x float> }
100+
; CHECK: %"RWBuffer<double>" = type { double }
101+
; CHECK: %"RWBuffer<int32_t4>" = type { <4 x i32> }
102102
; CHECK: %RWByteAddressBuffer = type { i32 }
103-
; CHECK: %RWStructuredBuffer = type { i16 }
104-
; CHECK: %RasterizerOrderedTypedBuffer = type { <4 x i32> }
105-
; CHECK: %RasterizerOrderedStructuredBuffer = type { <4 x i32> }
103+
; CHECK: %"RWStructuredBuffer<int16_t>" = type { i16 }
104+
; CHECK: %"RasterizerOrderedBuffer<int32_t4>" = type { <4 x i32> }
105+
; CHECK: %"RasterizerOrderedStructuredBuffer<int32_t4>" = type { <4 x i32> }
106106
; CHECK: %RasterizerOrderedByteAddressBuffer = type { i32 }
107-
; CHECK: %RWTypedBuffer.3 = type { i64 }
108-
; CHECK: %RWTypedBuffer.4 = type { <4 x float> }
109-
; CHECK: %RWTypedBuffer.5 = type { i64 }
110-
111-
; CHECK: @Zero = external constant %RWTypedBuffer
112-
; CHECK: @One = external constant %RWTypedBuffer.0
113-
; CHECK: @Two = external constant %RWTypedBuffer.1
114-
; CHECK: @Three = external constant %RWTypedBuffer.2
107+
; CHECK: %"RWBuffer<uint32_t>" = type { i64 }
108+
; CHECK: %"RWBuffer<float4>" = type { <4 x float> }
109+
110+
; CHECK: @Zero = external constant %"RWBuffer<half4>"
111+
; CHECK: @One = external constant %"RWBuffer<float2>"
112+
; CHECK: @Two = external constant %"RWBuffer<double>"
113+
; CHECK: @Three = external constant %"RWBuffer<int32_t4>"
115114
; CHECK: @Four = external constant %RWByteAddressBuffer
116-
; CHECK: @Five = external constant %RWStructuredBuffer
117-
; CHECK: @Six = external constant %RasterizerOrderedTypedBuffer
118-
; CHECK: @Seven = external constant %RasterizerOrderedStructuredBuffer
115+
; CHECK: @Five = external constant %"RWStructuredBuffer<int16_t>"
116+
; CHECK: @Six = external constant %"RasterizerOrderedBuffer<int32_t4>"
117+
; CHECK: @Seven = external constant %"RasterizerOrderedStructuredBuffer<int32_t4>"
119118
; CHECK: @Eight = external constant %RasterizerOrderedByteAddressBuffer
120-
; CHECK: @Nine = external constant %RWTypedBuffer.3
121-
; CHECK: @Array = external constant %RWTypedBuffer.4
122-
; CHECK: @Ten = external constant %RWTypedBuffer.5
119+
; CHECK: @Nine = external constant %"RWBuffer<uint32_t>"
120+
; CHECK: @Array = external constant %"RWBuffer<float4>"
121+
; CHECK: @Ten = external constant %"RWBuffer<uint32_t>"
123122

124123
; CHECK: !dx.resources = !{[[ResList:[!][0-9]+]]}
125124

0 commit comments

Comments
 (0)