From f52b4de3b29ee27213b7d60fb620a0d5d50b49f9 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Tue, 29 Nov 2022 18:17:33 -0800
Subject: Allow `no_diff` modifier on parameters (#2538)

---
 source/slang/slang-check-shader.cpp | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'source/slang/slang-check-shader.cpp')

diff --git a/source/slang/slang-check-shader.cpp b/source/slang/slang-check-shader.cpp
index a84e40768..3a64f3c8f 100644
--- a/source/slang/slang-check-shader.cpp
+++ b/source/slang/slang-check-shader.cpp
@@ -583,6 +583,17 @@ namespace Slang
         return varDecl->getName();
     }
 
+    Type* getParamType(ASTBuilder* astBuilder, DeclRef<VarDeclBase> const& paramDeclRef)
+    {
+        auto paramType = getType(astBuilder, paramDeclRef);
+        if (paramDeclRef.getDecl()->findModifier<NoDiffModifier>())
+        {
+            auto modifierVal = static_cast<Val*>(astBuilder->getOrCreate<NoDiffModifierVal>());
+            paramType = astBuilder->getModifiedType(paramType, 1, &modifierVal);
+        }
+        return paramType;
+    }
+
     void Module::_collectShaderParams()
     {
         auto moduleDecl = m_moduleDecl;
-- 
cgit v1.2.3