[DirectX] Validate if Textures/TypedBuffers are being bound in Root Signatures

llvm/lib/Target/DirectX/DXILPostOptimizationValidation.cpp

@@ -85,6 +85,17 @@ static void reportOverlappingBinding(Module &M, DXILResourceMap &DRM) {
   }
 }
 
+static void
+reportInvalidHandleTyBoundInRs(Module &M, Twine Type,
+                               ResourceInfo::ResourceBinding Binding) {
+  SmallString<128> Message;
+  raw_svector_ostream OS(Message);
+  OS << "resource " << Type << " at register (space=" << Binding.Space
+     << ", register=" << Binding.LowerBound << ")"
+     << " is bound to a texture or typed buffer.";
+  M.getContext().diagnose(DiagnosticInfoGeneric(Message));
+}
+
 static void reportRegNotBound(Module &M, Twine Type,
                               ResourceInfo::ResourceBinding Binding) {
   SmallString<128> Message;
@@ -198,7 +209,7 @@ getRootSignature(RootSignatureBindingInfo &RSBI,
   return RootSigDesc;
 }
 
-static void reportUnboundRegisters(
+static void reportInvalidRegistersBinding(
     Module &M,
     const llvm::ArrayRef<llvm::dxil::ResourceInfo::ResourceBinding> &Bindings,
     const iterator_range<SmallVectorImpl<dxil::ResourceInfo>::iterator>
@@ -217,6 +228,13 @@ static void reportUnboundRegisters(
     }
     if (!Bound) {
       reportRegNotBound(M, Res->getName(), Res->getBinding());
+    } else {
+      TargetExtType *Handle = Res->getHandleTy();
+      auto *TypedBuffer = dyn_cast_or_null<TypedBufferExtType>(Handle);
+      auto *Texture = dyn_cast_or_null<TextureExtType>(Handle);
+
+      if (TypedBuffer != nullptr || Texture != nullptr)
+        reportInvalidHandleTyBoundInRs(M, Res->getName(), Res->getBinding());
     }
   }
 }
@@ -239,16 +257,16 @@ static void reportErrors(Module &M, DXILResourceMap &DRM,
     RootSignatureBindingValidation Validation =
         initRSBindingValidation(*RSD, tripleToVisibility(MMI.ShaderProfile));
 
-    reportUnboundRegisters(
+    reportInvalidRegistersBinding(
         M, Validation.getBindingsOfType(dxbc::DescriptorRangeType::CBV),
         DRM.cbuffers());
-    reportUnboundRegisters(
+    reportInvalidRegistersBinding(
         M, Validation.getBindingsOfType(dxbc::DescriptorRangeType::UAV),
         DRM.uavs());
-    reportUnboundRegisters(
+    reportInvalidRegistersBinding(
         M, Validation.getBindingsOfType(dxbc::DescriptorRangeType::Sampler),
         DRM.samplers());
-    reportUnboundRegisters(
+    reportInvalidRegistersBinding(
         M, Validation.getBindingsOfType(dxbc::DescriptorRangeType::SRV),
         DRM.srvs());
   }

llvm/test/CodeGen/DirectX/rootsignature-validation-textures-fail.ll

@@ -0,0 +1,31 @@
+; RUN: not opt -S -passes='dxil-post-optimization-validation' -mtriple=dxil-pc-shadermodel6.6-compute %s 2>&1 | FileCheck %s
+; CHECK: error: resource TB at register (space=0, register=0) is bound to a texture or typed buffer.
+
+
+; Root Signature(
+;   CBV(b3, space=1, visibility=SHADER_VISIBILITY_ALL)
+;   DescriptorTable(SRV(t0, space=0, numDescriptors=1), visibility=SHADER_VISIBILITY_ALL)
+;   DescriptorTable(Sampler(s0, numDescriptors=2), visibility=SHADER_VISIBILITY_VERTEX)
+;   DescriptorTable(UAV(u0, numDescriptors=unbounded), visibility=SHADER_VISIBILITY_ALL)
+
+@TB.str = private unnamed_addr constant [3 x i8] c"TB\00", align 1
+
+define void @CSMain() "hlsl.shader"="compute" {
+entry:
+
+  %TB =  tail call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding.tdx.TypedBuffer_f32_1_0_0t(i32 0, i32 0, i32 1, i32 0, i1 false, ptr nonnull @TB.str)
+
+  ret void
+}
+
+!dx.rootsignatures = !{!0}
+
+!0 = !{ptr @CSMain, !1, i32 2}
+!1 = !{!2, !3, !5, !7}
+!2 = !{!"RootCBV", i32 0, i32 3, i32 1, i32 4}
+!3 = !{!"DescriptorTable", i32 0, !4}
+!4 = !{!"SRV", i32 1, i32 0, i32 0, i32 -1, i32 0}
+!5 = !{!"DescriptorTable", i32 0, !6}
+!6 = !{!"Sampler", i32 5, i32 3, i32 2, i32 -1, i32 0}
+!7 = !{!"DescriptorTable", i32 0, !8}
+!8 = !{!"UAV", i32 -1, i32 0, i32 0, i32 -1, i32 2}

llvm/test/CodeGen/DirectX/rootsignature-validation.ll

@@ -21,7 +21,7 @@ entry:
   %CB = tail call target("dx.CBuffer", target("dx.Layout", %__cblayout_CB, 4, 0)) @llvm.dx.resource.handlefrombinding(i32 1, i32 3, i32 1, i32 0, i1 false, ptr nonnull @CB.str)
   %Sampler = call target("dx.Sampler", 0) @llvm.dx.resource.handlefrombinding(i32 2, i32 3, i32 1, i32 0, i1 false, ptr nonnull @Smp.str)
   %SB = tail call target("dx.RawBuffer", i32, 0, 0) @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i32_0_0t(i32 0, i32 0, i32 1, i32 0, i1 false, ptr nonnull @SB.str)
-  %RWB =  tail call target("dx.TypedBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding.tdx.TypedBuffer_f32_1_0_0t(i32 0, i32 0, i32 1, i32 0, i1 false, ptr nonnull @RWB.str)
+  %RWB =  tail call target("dx.RawBuffer", float, 1, 0, 0) @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f32_1_0_0t(i32 0, i32 0, i32 1, i32 0, i1 false, ptr nonnull @RWB.str)
 
   ret void
 }