From f20f4e2b39142dfab55be05eea064033719665ad Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Fri, 29 Sep 2023 12:01:36 -0700
Subject: Add `requirePrelude()` intrinsic function. (#3250)

* Add `requirePrelude()` intrinsic function.

* Fix.

---------

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

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

diff --git a/source/slang/slang-emit-c-like.cpp b/source/slang/slang-emit-c-like.cpp
index fb216ae01..a2c3abb5e 100644
--- a/source/slang/slang-emit-c-like.cpp
+++ b/source/slang/slang-emit-c-like.cpp
@@ -112,6 +112,15 @@ void CLikeSourceEmitter::emitFrontMatterImpl(TargetRequest* targetReq)
     SLANG_UNUSED(targetReq);
 }
 
+void CLikeSourceEmitter::emitPreModuleImpl()
+{
+    for (auto prelude : m_requiredPreludes)
+    {
+        m_writer->emit(prelude->getStringSlice());
+        m_writer->emit("\n");
+    }
+}
+
 //
 // Types
 //
@@ -1856,6 +1865,22 @@ void CLikeSourceEmitter::emitCallExpr(IRCall* inst, EmitOpInfo outerPrec)
     UnownedStringSlice intrinsicDefinition;
     if (findTargetIntrinsicDefinition(funcValue, intrinsicDefinition))
     {
+        // Make sure we register all required preludes for emit.
+        if (auto func = as<IRFunc>(getResolvedInstForDecorations(funcValue)))
+        {
+            for (auto block : func->getBlocks())
+            {
+                for (auto ii : block->getChildren())
+                {
+                    if (auto requirePrelude = as<IRRequirePrelude>(ii))
+                    {
+                        auto preludeTextInst = as<IRStringLit>(requirePrelude->getOperand(0));
+                        if (preludeTextInst)
+                            m_requiredPreludes.add(preludeTextInst);
+                    }
+                }
+            }
+        }
         emitIntrinsicCallExpr(inst, intrinsicDefinition, outerPrec);
     }
     else
-- 
cgit v1.2.3