aboutsummaryrefslogtreecommitdiff
path: root/llvm/unittests/Target/DirectX/ResourceBindingAnalysisTests.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'llvm/unittests/Target/DirectX/ResourceBindingAnalysisTests.cpp')
-rw-r--r--llvm/unittests/Target/DirectX/ResourceBindingAnalysisTests.cpp171
1 files changed, 7 insertions, 164 deletions
diff --git a/llvm/unittests/Target/DirectX/ResourceBindingAnalysisTests.cpp b/llvm/unittests/Target/DirectX/ResourceBindingAnalysisTests.cpp
index 6cd1a48..d4715be 100644
--- a/llvm/unittests/Target/DirectX/ResourceBindingAnalysisTests.cpp
+++ b/llvm/unittests/Target/DirectX/ResourceBindingAnalysisTests.cpp
@@ -44,19 +44,6 @@ protected:
delete MAM;
delete Context;
}
-
- void checkExpectedSpaceAndFreeRanges(
- DXILResourceBindingInfo::RegisterSpace &RegSpace, uint32_t ExpSpace,
- ArrayRef<uint32_t> ExpValues) {
- EXPECT_EQ(RegSpace.Space, ExpSpace);
- EXPECT_EQ(RegSpace.FreeRanges.size() * 2, ExpValues.size());
- unsigned I = 0;
- for (auto &R : RegSpace.FreeRanges) {
- EXPECT_EQ(R.LowerBound, ExpValues[I]);
- EXPECT_EQ(R.UpperBound, ExpValues[I + 1]);
- I += 2;
- }
- }
};
TEST_F(ResourceBindingAnalysisTest, TestTrivialCase) {
@@ -76,103 +63,16 @@ entry:
EXPECT_EQ(false, DRBI.hasImplicitBinding());
EXPECT_EQ(false, DRBI.hasOverlappingBinding());
-
- // check that UAV has exactly one gap
- DXILResourceBindingInfo::BindingSpaces &UAVSpaces =
- DRBI.getBindingSpaces(ResourceClass::UAV);
- EXPECT_EQ(UAVSpaces.RC, ResourceClass::UAV);
- EXPECT_EQ(UAVSpaces.Spaces.size(), 1u);
- checkExpectedSpaceAndFreeRanges(UAVSpaces.Spaces[0], 0,
- {0, 4, 6, UINT32_MAX});
-
- // check that other kinds of register spaces are all available
- for (auto RC :
- {ResourceClass::SRV, ResourceClass::CBuffer, ResourceClass::Sampler}) {
- DXILResourceBindingInfo::BindingSpaces &Spaces = DRBI.getBindingSpaces(RC);
- EXPECT_EQ(Spaces.RC, RC);
- EXPECT_EQ(Spaces.Spaces.size(), 0u);
- }
-}
-
-TEST_F(ResourceBindingAnalysisTest, TestManyBindings) {
- // cbuffer CB : register(b3) { int a; }
- // RWBuffer<float4> A[5] : register(u10, space20);
- // StructuredBuffer<int> B : register(t5);
- // RWBuffer<float> C : register(u5);
- // StructuredBuffer<int> D[5] : register(t0);
- // RWBuffer<float> E[2] : register(u2);
- // SamplerState S1 : register(s5, space2);
- // SamplerState S2 : register(s4, space2);
- StringRef Assembly = R"(
-%__cblayout_CB = type <{ i32 }>
-define void @main() {
-entry:
- %handleCB = call target("dx.CBuffer", target("dx.Layout", %__cblayout_CB, 4, 0)) @llvm.dx.resource.handlefrombinding(i32 0, i32 3, i32 1, i32 0, i1 false, ptr null)
- %handleA = call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding(i32 20, i32 10, i32 5, i32 0, i1 false, ptr null)
- %handleB = call target("dx.RawBuffer", i32, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 5, i32 1, i32 0, i1 false, ptr null)
- %handleC = call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 5, i32 1, i32 0, i1 false, ptr null)
- %handleD = call target("dx.RawBuffer", i32, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 5, i32 4, i1 false, ptr null)
- %handleE = call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 2, i32 2, i32 0, i1 false, ptr null)
- %handleS1 = call target("dx.Sampler", 0) @llvm.dx.resource.handlefrombinding(i32 2, i32 5, i32 1, i32 0, i1 false, ptr null)
- %handleS2 = call target("dx.Sampler", 0) @llvm.dx.resource.handlefrombinding(i32 2, i32 4, i32 1, i32 0, i1 false, ptr null)
- ; duplicate binding for the same resource
- %handleD2 = call target("dx.RawBuffer", i32, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 5, i32 4, i1 false, ptr null)
- ret void
-}
- )";
-
- auto M = parseAsm(Assembly);
-
- DXILResourceBindingInfo &DRBI =
- MAM->getResult<DXILResourceBindingAnalysis>(*M);
-
- EXPECT_EQ(false, DRBI.hasImplicitBinding());
- EXPECT_EQ(false, DRBI.hasOverlappingBinding());
-
- DXILResourceBindingInfo::BindingSpaces &SRVSpaces =
- DRBI.getBindingSpaces(ResourceClass::SRV);
- EXPECT_EQ(SRVSpaces.RC, ResourceClass::SRV);
- EXPECT_EQ(SRVSpaces.Spaces.size(), 1u);
- // verify that consecutive bindings are merged
- // (SRVSpaces has only one free space range {6, UINT32_MAX}).
- checkExpectedSpaceAndFreeRanges(SRVSpaces.Spaces[0], 0, {6, UINT32_MAX});
-
- DXILResourceBindingInfo::BindingSpaces &UAVSpaces =
- DRBI.getBindingSpaces(ResourceClass::UAV);
- EXPECT_EQ(UAVSpaces.RC, ResourceClass::UAV);
- EXPECT_EQ(UAVSpaces.Spaces.size(), 2u);
- checkExpectedSpaceAndFreeRanges(UAVSpaces.Spaces[0], 0,
- {0, 1, 4, 4, 6, UINT32_MAX});
- checkExpectedSpaceAndFreeRanges(UAVSpaces.Spaces[1], 20,
- {0, 9, 15, UINT32_MAX});
-
- DXILResourceBindingInfo::BindingSpaces &CBufferSpaces =
- DRBI.getBindingSpaces(ResourceClass::CBuffer);
- EXPECT_EQ(CBufferSpaces.RC, ResourceClass::CBuffer);
- EXPECT_EQ(CBufferSpaces.Spaces.size(), 1u);
- checkExpectedSpaceAndFreeRanges(CBufferSpaces.Spaces[0], 0,
- {0, 2, 4, UINT32_MAX});
-
- DXILResourceBindingInfo::BindingSpaces &SamplerSpaces =
- DRBI.getBindingSpaces(ResourceClass::Sampler);
- EXPECT_EQ(SamplerSpaces.RC, ResourceClass::Sampler);
- EXPECT_EQ(SamplerSpaces.Spaces.size(), 1u);
- checkExpectedSpaceAndFreeRanges(SamplerSpaces.Spaces[0], 2,
- {0, 3, 6, UINT32_MAX});
}
-TEST_F(ResourceBindingAnalysisTest, TestUnboundedAndOverlap) {
- // StructuredBuffer<float> A[] : register(t5);
- // StructuredBuffer<float> B[3] : register(t0);
- // StructuredBuffer<float> C[] : register(t0, space2);
- // StructuredBuffer<float> D : register(t4, space2); /* overlapping */
+TEST_F(ResourceBindingAnalysisTest, TestOverlap) {
+ // StructuredBuffer<float> A[] : register(t0, space2);
+ // StructuredBuffer<float> B : register(t4, space2); /* overlapping */
StringRef Assembly = R"(
define void @main() {
entry:
- %handleA = call target("dx.RawBuffer", float, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 5, i32 -1, i32 10, i1 false, ptr null)
- %handleB = call target("dx.RawBuffer", float, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 3, i32 0, i1 false, ptr null)
- %handleC = call target("dx.RawBuffer", float, 0, 0) @llvm.dx.resource.handlefrombinding(i32 2, i32 0, i32 -1, i32 100, i1 false, ptr null)
- %handleD = call target("dx.RawBuffer", float, 0, 0) @llvm.dx.resource.handlefrombinding(i32 2, i32 4, i32 1, i32 0, i1 false, ptr null)
+ %handleA = call target("dx.RawBuffer", float, 0, 0) @llvm.dx.resource.handlefrombinding(i32 2, i32 0, i32 -1, i32 100, i1 false, ptr null)
+ %handleB = call target("dx.RawBuffer", float, 0, 0) @llvm.dx.resource.handlefrombinding(i32 2, i32 4, i32 1, i32 0, i1 false, ptr null)
ret void
}
)";
@@ -184,13 +84,6 @@ entry:
EXPECT_EQ(false, DRBI.hasImplicitBinding());
EXPECT_EQ(true, DRBI.hasOverlappingBinding());
-
- DXILResourceBindingInfo::BindingSpaces &SRVSpaces =
- DRBI.getBindingSpaces(ResourceClass::SRV);
- EXPECT_EQ(SRVSpaces.RC, ResourceClass::SRV);
- EXPECT_EQ(SRVSpaces.Spaces.size(), 2u);
- checkExpectedSpaceAndFreeRanges(SRVSpaces.Spaces[0], 0, {3, 4});
- checkExpectedSpaceAndFreeRanges(SRVSpaces.Spaces[1], 2, {});
}
TEST_F(ResourceBindingAnalysisTest, TestExactOverlap) {
@@ -214,49 +107,6 @@ entry:
EXPECT_EQ(false, DRBI.hasImplicitBinding());
EXPECT_EQ(true, DRBI.hasOverlappingBinding());
-
- DXILResourceBindingInfo::BindingSpaces &SRVSpaces =
- DRBI.getBindingSpaces(ResourceClass::SRV);
- EXPECT_EQ(SRVSpaces.RC, ResourceClass::SRV);
- EXPECT_EQ(SRVSpaces.Spaces.size(), 1u);
- checkExpectedSpaceAndFreeRanges(SRVSpaces.Spaces[0], 0,
- {0, 4, 6, UINT32_MAX});
-}
-
-TEST_F(ResourceBindingAnalysisTest, TestEndOfRange) {
- // RWBuffer<float> A : register(u4294967295); /* UINT32_MAX */
- // RWBuffer<float> B[10] : register(u4294967286, space1);
- // /* range (UINT32_MAX - 9, UINT32_MAX )*/
- // RWBuffer<float> C[10] : register(u2147483647, space2);
- // /* range (INT32_MAX, INT32_MAX + 9) */
- StringRef Assembly = R"(
-%__cblayout_CB = type <{ i32 }>
-define void @main() {
-entry:
- %handleA = call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 -1, i32 1, i32 0, i1 false, ptr null)
- %handleB = call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding(i32 1, i32 -10, i32 10, i32 50, i1 false, ptr null)
- %handleC = call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding(i32 2, i32 2147483647, i32 10, i32 100, i1 false, ptr null)
- ret void
-}
- )";
-
- auto M = parseAsm(Assembly);
-
- DXILResourceBindingInfo &DRBI =
- MAM->getResult<DXILResourceBindingAnalysis>(*M);
-
- EXPECT_EQ(false, DRBI.hasImplicitBinding());
- EXPECT_EQ(false, DRBI.hasOverlappingBinding());
-
- DXILResourceBindingInfo::BindingSpaces &UAVSpaces =
- DRBI.getBindingSpaces(ResourceClass::UAV);
- EXPECT_EQ(UAVSpaces.RC, ResourceClass::UAV);
- EXPECT_EQ(UAVSpaces.Spaces.size(), 3u);
- checkExpectedSpaceAndFreeRanges(UAVSpaces.Spaces[0], 0, {0, UINT32_MAX - 1});
- checkExpectedSpaceAndFreeRanges(UAVSpaces.Spaces[1], 1, {0, UINT32_MAX - 10});
- checkExpectedSpaceAndFreeRanges(
- UAVSpaces.Spaces[2], 2,
- {0, (uint32_t)INT32_MAX - 1, (uint32_t)INT32_MAX + 10, UINT32_MAX});
}
TEST_F(ResourceBindingAnalysisTest, TestImplicitFlag) {
@@ -275,15 +125,8 @@ entry:
DXILResourceBindingInfo &DRBI =
MAM->getResult<DXILResourceBindingAnalysis>(*M);
- EXPECT_EQ(true, DRBI.hasImplicitBinding());
- EXPECT_EQ(false, DRBI.hasOverlappingBinding());
-
- DXILResourceBindingInfo::BindingSpaces &UAVSpaces =
- DRBI.getBindingSpaces(ResourceClass::UAV);
- EXPECT_EQ(UAVSpaces.RC, ResourceClass::UAV);
- EXPECT_EQ(UAVSpaces.Spaces.size(), 1u);
- checkExpectedSpaceAndFreeRanges(UAVSpaces.Spaces[0], 100,
- {0, 4, 6, UINT32_MAX});
+ EXPECT_TRUE(DRBI.hasImplicitBinding());
+ EXPECT_FALSE(DRBI.hasOverlappingBinding());
}
} // namespace
generated by cgit v1.1 at 2025-08-01 22:06:58 +0000