From 332f60c19336252d907b83882aa70665ca93a9d2 Mon Sep 17 00:00:00 2001 From: Ellie Hermaszewska 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 --- source/slang/slang-ir-specialize-function-call.cpp | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'source/slang/slang-ir-specialize-function-call.cpp') diff --git a/source/slang/slang-ir-specialize-function-call.cpp b/source/slang/slang-ir-specialize-function-call.cpp index 9b7cdaea6..bc238e6ec 100644 --- a/source/slang/slang-ir-specialize-function-call.cpp +++ b/source/slang/slang-ir-specialize-function-call.cpp @@ -31,6 +31,9 @@ bool FunctionCallSpecializeCondition::isParamSuitableForSpecialization(IRParam* // if (as(arg)) return true; + // Similarly for these global values + if( as(arg) ) return true; + // As we will see later, we can also // specialize a call when the argument // is the result of indexing into an @@ -505,6 +508,11 @@ struct FunctionParameterSpecializationContext // ioInfo.key.vals.add(oldGlobalParam); } + else if( auto globalConstant = as(oldArg) ) + { + // Similarly for other global constants + ioInfo.key.vals.add(globalConstant); + } else if( oldArg->getOp() == kIROp_GetElement ) { // This is the case where the `oldArg` is @@ -626,6 +634,12 @@ struct FunctionParameterSpecializationContext // return globalParam; } + if( auto globalFunc = as(oldArg) ) + { + // As above, the identity of the specialized function is sufficient + // to resolve the uses + return globalFunc; + } else if( oldArg->getOp() == kIROp_GetElement ) { // This is the case where the argument is -- cgit v1.2.3