Support SM6.6 keyword "WaveSize" (#3871)

Resolves an issue #3385

Shader Model 6.6 added a new keyowrd, "WaveSize". See the following link
for more details:
  https://microsoft.github.io/DirectX-Specs/d3d/HLSL_SM_6_6_WaveSize.html

Co-authored-by: Yong He <yonghe@outlook.com>

1 files changed, 10 insertions, 0 deletions

diff --git a/source/slang/slang-emit-c-like.cpp b/source/slang/slang-emit-c-like.cpp
index 5813819b4..58ca39b69 100644
--- a/source/slang/slang-emit-c-like.cpp
+++ b/source/slang/slang-emit-c-like.cpp
@@ -268,6 +268,16 @@ void CLikeSourceEmitter::emitSimpleType(IRType* type)
     return decor;
 }
 
+/* static */IRWaveSizeDecoration* CLikeSourceEmitter::getComputeWaveSize(IRFunc* func, Int* outWaveSize)
+{
+    IRWaveSizeDecoration* decor = func->findDecoration<IRWaveSizeDecoration>();
+    if (decor)
+    {
+        *outWaveSize = Int(getIntVal(decor->getOperand(0)));
+    }
+    return decor;
+}
+
 List<IRWitnessTableEntry*> CLikeSourceEmitter::getSortedWitnessTableEntries(IRWitnessTable* witnessTable)
 {
     List<IRWitnessTableEntry*> sortedWitnessTableEntries;