From f44da6cc5c0f211c13bd1eb0743d79c7861ea64e Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Thu, 8 Feb 2024 18:29:32 -0800
Subject: Support pointers in SPIRV. (#3561)

* Support pointers in SPIRV.

* Fix test.

* Enhance test.

* Fix test.

* Cleanup.
---
 source/slang/slang-ir-util.cpp | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'source/slang/slang-ir-util.cpp')

diff --git a/source/slang/slang-ir-util.cpp b/source/slang/slang-ir-util.cpp
index d859f86a6..f514fea1d 100644
--- a/source/slang/slang-ir-util.cpp
+++ b/source/slang/slang-ir-util.cpp
@@ -1030,6 +1030,26 @@ IRInst* getInstInBlock(IRInst* inst)
     return getInstInBlock(inst->getParent());
 }
 
+ShortList<IRInst*> getPhiArgs(IRInst* phiParam)
+{
+    ShortList<IRInst*> result;
+    auto block = cast<IRBlock>(phiParam->getParent());
+    UInt paramIndex = 0;
+    for (auto p = block->getFirstParam(); p; p = p->getNextParam())
+    {
+        if (p == phiParam)
+            break;
+        paramIndex++;
+    }
+    for (auto predBlock : block->getPredecessors())
+    {
+        auto termInst = as<IRUnconditionalBranch>(predBlock->getTerminator());
+        SLANG_ASSERT(paramIndex < termInst->getArgCount());
+        result.add(termInst->getArg(paramIndex));
+    }
+    return result;
+}
+
 void removePhiArgs(IRInst* phiParam)
 {
     auto block = cast<IRBlock>(phiParam->getParent());
-- 
cgit v1.2.3