More direct-SPIRV fixes. (#3257)

* More direct-SPIRV fixes.

* Fix array-reg-to-mem.

---------

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 14 insertions, 0 deletions

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);