From a85c350df03c6cdf9b433f58fc0e66affda03e9e Mon Sep 17 00:00:00 2001
From: Darren Wihandi <65404740+fairywreath@users.noreply.github.com>
Date: Sat, 18 Jan 2025 02:07:16 -0500
Subject: Implement Quad Control intrinsics (#5981)

---
 source/slang/slang-emit-spirv.cpp | 45 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 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 2cf84a854..951507ab9 100644
--- a/source/slang/slang-emit-spirv.cpp
+++ b/source/slang/slang-emit-spirv.cpp
@@ -3544,6 +3544,37 @@ struct SPIRVEmitContext : public SourceEmitterBase, public SPIRVEmitSharedContex
 
                 break;
             }
+
+        case kIROp_RequireMaximallyReconverges:
+            if (auto entryPointsUsingInst =
+                    getReferencingEntryPoints(m_referencingEntryPoints, getParentFunc(inst)))
+            {
+                ensureExtensionDeclaration(UnownedStringSlice("SPV_KHR_maximal_reconvergence"));
+                for (IRFunc* entryPoint : *entryPointsUsingInst)
+                {
+                    requireSPIRVExecutionMode(
+                        nullptr,
+                        getIRInstSpvID(entryPoint),
+                        SpvExecutionModeMaximallyReconvergesKHR);
+                }
+            }
+            break;
+        case kIROp_RequireQuadDerivatives:
+            if (auto entryPointsUsingInst =
+                    getReferencingEntryPoints(m_referencingEntryPoints, getParentFunc(inst)))
+            {
+                ensureExtensionDeclaration(UnownedStringSlice("SPV_KHR_quad_control"));
+                requireSPIRVCapability(SpvCapabilityQuadControlKHR);
+                for (IRFunc* entryPoint : *entryPointsUsingInst)
+                {
+                    requireSPIRVExecutionMode(
+                        nullptr,
+                        getIRInstSpvID(entryPoint),
+                        SpvExecutionModeQuadDerivativesKHR);
+                }
+            }
+            break;
+
         case kIROp_Return:
             if (as<IRReturn>(inst)->getVal()->getOp() == kIROp_VoidLit)
                 result = emitOpReturn(parent, inst);
@@ -4472,6 +4503,20 @@ struct SPIRVEmitContext : public SourceEmitterBase, public SPIRVEmitSharedContex
                 }
             }
             break;
+        case kIROp_MaximallyReconvergesDecoration:
+            ensureExtensionDeclaration(UnownedStringSlice("SPV_khr_maximal_reconvergence"));
+            requireSPIRVExecutionMode(nullptr, dstID, SpvExecutionModeMaximallyReconvergesKHR);
+            break;
+        case kIROp_QuadDerivativesDecoration:
+            ensureExtensionDeclaration(UnownedStringSlice("SPV_KHR_quad_control"));
+            requireSPIRVCapability(SpvCapabilityQuadControlKHR);
+            requireSPIRVExecutionMode(nullptr, dstID, SpvExecutionModeQuadDerivativesKHR);
+            break;
+        case kIROp_RequireFullQuadsDecoration:
+            ensureExtensionDeclaration(UnownedStringSlice("SPV_KHR_quad_control"));
+            requireSPIRVCapability(SpvCapabilityQuadControlKHR);
+            requireSPIRVExecutionMode(nullptr, dstID, SpvExecutionModeRequireFullQuadsKHR);
+            break;
         case kIROp_SPIRVBufferBlockDecoration:
             {
                 emitOpDecorate(
-- 
cgit v1.2.3