Add a check to ensure entry shader stage is the same as target profile

for non-library target profile compilation.

Add a test to verify the check triggers diagnostic report appropriately.

Author: bharadwajy

llvm/lib/Target/DirectX/DXILTranslateMetadata.cpp

@@ -330,6 +330,13 @@ static void translateMetadata(Module &M, const DXILResourceMap &DRM,
     uint64_t EntryShaderFlags =
         (MMDI.ShaderProfile == Triple::EnvironmentType::Library) ? 0
                                                                  : ShaderFlags;
+    if (MMDI.ShaderProfile != Triple::EnvironmentType::Library) {
+      if (EntryProp.ShaderStage != MMDI.ShaderProfile) {
+        M.getContext().diagnose(DiagnosticInfoModuleFormat(
+            M, "Non-library shader: Stage of Shader entry different from "
+               "target profile"));
+      }
+    }
     EntryFnMDNodes.emplace_back(emitEntryMD(EntryProp, Signatures, ResourceMD,
                                             EntryShaderFlags,
                                             MMDI.ShaderProfile));

llvm/test/CodeGen/DirectX/Metadata/target-profile-error.ll

@@ -0,0 +1,12 @@
+; RUN: not opt -S -dxil-translate-metadata %s 2>&1 | FileCheck %s
+
+target triple = "dxil-pc-shadermodel6.6-pixel"
+
+; CHECK: Non-library shader: Stage of Shader entry different from target profile
+
+define void @entry() #0 {
+entry:
+  ret void
+}
+
+attributes #0 = { noinline nounwind "exp-shader"="cs" "hlsl.numthreads"="1,2,1" "hlsl.shader"="compute" }