From 332f60c19336252d907b83882aa70665ca93a9d2 Mon Sep 17 00:00:00 2001
From: Ellie Hermaszewska <ellieh@nvidia.com>
Date: Fri, 12 May 2023 04:33:21 +0800
Subject: MVP for higher order functions (#2849)

* MVP for higher order functions

* Add shader subgroup partitioned glsl intrinsics

* Implement parsing and checking for tuple types

Currently there is no way to do anything useful with them from the source language however

* neaten

* Correct precedence of function type parsing

* neaten

* higher order function tests

* function types of any arity

* Inference for higher order functions

* Add second test for unsynchronized params

* regenerate vs projects

* dx11 -> dx12 for saturated cooperations tests

* Disable saturated cooperation tests on vulkan

They fail on release builds in CI, not essential for the higher order function work however

* remove saturated-cooperation tests

* Remove unnecessary assert and clarify control flow in AddDeclRefOverloadCandidates

* Add Tuple type name mangling

* Use functype keyword to introduce function types

* Add more inference tests for hof

---------

Co-authored-by: Yong He <yonghe@outlook.com>
---
 source/slang/slang-ir-defunctionalization.cpp | 39 +++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 source/slang/slang-ir-defunctionalization.cpp

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

diff --git a/source/slang/slang-ir-defunctionalization.cpp b/source/slang/slang-ir-defunctionalization.cpp
new file mode 100644
index 000000000..ac4258753
--- /dev/null
+++ b/source/slang/slang-ir-defunctionalization.cpp
@@ -0,0 +1,39 @@
+#include "slang-ir-defunctionalization.h"
+
+#include "slang-ir-insts.h"
+#include "slang-ir-specialize-function-call.h"
+#include "slang-ir-ssa-simplification.h"
+#include "slang-ir.h"
+
+namespace Slang
+{
+
+struct FunctionParameterSpecializationCondition : FunctionCallSpecializeCondition
+{
+    TargetRequest* targetRequest = nullptr;
+
+    bool doesParamWantSpecialization(IRParam* param, IRInst* /*arg*/)
+    {
+        IRType* type = param->getDataType();
+        return as<IRFuncType>(type);
+    }
+};
+
+bool specializeHigherOrderParameters(
+    CodeGenContext* codeGenContext,
+    IRModule*       module)
+{
+    bool result = false;
+    FunctionParameterSpecializationCondition condition;
+    condition.targetRequest = codeGenContext->getTargetReq();
+    bool changed = true;
+    while (changed)
+    {
+        changed = specializeFunctionCalls(codeGenContext, module, &condition);
+        simplifyIR(module);
+        result |= changed;
+    }
+    return result;
+}
+
+} // namespace Slang
-- 
cgit v1.2.3