From d87493a46c00be37b820a473c0827bbb865eb222 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Mon, 2 Oct 2023 15:39:34 -0700
Subject: More direct-SPIRV fixes. (#3257)

* More direct-SPIRV fixes.

* Fix array-reg-to-mem.

---------

Co-authored-by: Yong He <yhe@nvidia.com>
---
 source/slang/slang-emit-spirv.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'source/slang/slang-emit-spirv.cpp')

diff --git a/source/slang/slang-emit-spirv.cpp b/source/slang/slang-emit-spirv.cpp
index 104fab339..62e7d428b 100644
--- a/source/slang/slang-emit-spirv.cpp
+++ b/source/slang/slang-emit-spirv.cpp
@@ -2274,6 +2274,20 @@ struct SPIRVEmitContext
                 registerInst(inst, inner);
                 return inner;
             }
+        case kIROp_GetVulkanRayTracingPayloadLocation:
+            {
+                IRInst* location = getVulkanPayloadLocation(inst->getOperand(0));
+                if (!location)
+                {
+                    SLANG_DIAGNOSE_UNEXPECTED(m_sink, inst, "no payload location assigned.");
+                    IRBuilder builder(inst);
+                    builder.setInsertBefore(inst);
+                    location = builder.getIntValue(builder.getIntType(), 0);
+                }
+                auto inner = ensureInst(location);
+                registerInst(inst, inner);
+                return inner;
+            }
         case kIROp_Return:
             if (as<IRReturn>(inst)->getVal()->getOp() == kIROp_VoidLit)
                 return emitOpReturn(parent, inst);
-- 
cgit v1.2.3