source/slang/slang-ir-defunctionalization.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

#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*/, IRCall* /*callInst*/)
    {
        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);
        result |= changed;
    }
    return result;
}

} // namespace Slang