1 files changed, 14 insertions, 0 deletions
diff --git a/source/slang/slang-check-modifier.cpp b/source/slang/slang-check-modifier.cpp
index eae993dc7..17215d284 100644
--- a/source/slang/slang-check-modifier.cpp
+++ b/source/slang/slang-check-modifier.cpp
@@ -707,6 +707,13 @@ namespace Slang
 
             rayPayloadAttr->location = (int32_t)val->getValue();
         }
+        else if (auto rayPayloadInAttr = as<VulkanRayPayloadInAttribute>(attr))
+        {
+            SLANG_ASSERT(attr->args.getCount() == 1);
+            auto val = checkConstantIntVal(attr->args[0]);
+            if (!val) return false;
+            rayPayloadInAttr->location = (int32_t)val->getValue();
+        }
         else if (auto callablePayloadAttr = as<VulkanCallablePayloadAttribute>(attr))
         {
             SLANG_ASSERT(attr->args.getCount() == 1);
@@ -716,6 +723,13 @@ namespace Slang
 
             callablePayloadAttr->location = (int32_t)val->getValue();
         }
+        else if (auto callablePayloadInAttr = as<VulkanCallablePayloadInAttribute>(attr))
+        {
+            SLANG_ASSERT(attr->args.getCount() == 1);
+            auto val = checkConstantIntVal(attr->args[0]);
+            if (!val) return false;
+            callablePayloadInAttr->location = (int32_t)val->getValue();
+        }
         else if (auto hitObjectAttributesAttr = as<VulkanHitObjectAttributesAttribute>(attr))
         {
             SLANG_ASSERT(attr->args.getCount() == 1);