[llvm] [HLSL] Implement DXILResourceBindingAnalysis (PR #137258)

[Helena Kotas via llvm-commits]

================
@@ -0,0 +1,268 @@
+//===- llvm/unittests/Target/DirectX/ResourceBindingAnalysisTests.cpp -----===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/ADT/ArrayRef.h"
+#include "llvm/Analysis/DXILResource.h"
+#include "llvm/AsmParser/Parser.h"
+#include "llvm/Passes/PassBuilder.h"
+#include "llvm/Support/DXILABI.h"
+#include "gtest/gtest.h"
+#include <cstdint>
+
+using namespace llvm;
+using namespace llvm::dxil;
+
+namespace {
+class ResourceBindingAnalysisTest : public testing::Test {
+protected:
+  PassBuilder *PB;
+  ModuleAnalysisManager *MAM;
+  LLVMContext *Context;
+
+  virtual void SetUp() {
+    PB = new PassBuilder();
+    MAM = new ModuleAnalysisManager();
+    Context = new LLVMContext();
+    PB->registerModuleAnalyses(*MAM);
+    MAM->registerPass([&] { return DXILResourceBindingAnalysis(); });
+  }
+
+  std::unique_ptr<Module> parseAsm(StringRef Asm) {
+    SMDiagnostic Error;
+    std::unique_ptr<Module> M = parseAssemblyString(Asm, Error, *Context);
+    EXPECT_TRUE(M) << "Bad assembly?: " << Error.getMessage();
+    return M;
+  }
+
+  virtual void TearDown() {
+    delete PB;
+    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) {
+  // RWBuffer<float> Buf : register(u5);
+  StringRef Assembly = R"(
+define void @main() {
+entry:
+  %handle = call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding(i32 0, i32 5, i32 1, i32 0, i1 false)
+  
+  call void @a.func(target("dx.TypedBuffer", float, 1, 0, 0) %handle)
----------------
hekota wrote:

You are correct, this is not needed. :)

https://github.com/llvm/llvm-project/pull/137258