summaryrefslogtreecommitdiffstats
path: root/source/slang/slang-ir-missing-return.cpp
blob: 82d1171e050a45f9e65232bef68a9e09acc64191 (plain)
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// ir-missing-return.cpp
#include "slang-ir-missing-return.h"

#include "core/slang-type-text-util.h"
#include "slang-compiler.h"
#include "slang-ir-insts.h"
#include "slang-ir.h"

namespace Slang
{

class DiagnosticSink;
struct IRModule;

// Returns false if compilation target does not allow and errors out(i.e. during downstream
// compilation) on missing returns.
static bool doesTargetAllowMissingReturns(CodeGenTarget target)
{
    if (isKhronosTarget(target) || isWGPUTarget(target))
    {
        return false;
    }

    return true;
}

static void diagnoseMissingReturnForTarget(
    IRMissingReturn* missingReturn,
    DiagnosticSink* sink,
    CodeGenTarget target,
    bool diagnoseWarning)
{
    if (doesTargetAllowMissingReturns(target))
    {
        if (diagnoseWarning)
        {
            sink->diagnose(missingReturn, Diagnostics::missingReturn);
        }
    }
    else
    {
        sink->diagnose(
            missingReturn,
            Diagnostics::missingReturnError,
            TypeTextUtil::getCompileTargetName(SlangCompileTarget(target)));
    }
}

void checkForMissingReturnsRec(
    IRInst* inst,
    DiagnosticSink* sink,
    CodeGenTarget target,
    bool diagnoseWarning)
{
    if (auto code = as<IRGlobalValueWithCode>(inst))
    {
        for (auto block : code->getBlocks())
        {
            auto terminator = block->getTerminator();

            if (auto missingReturn = as<IRMissingReturn>(terminator))
            {
                diagnoseMissingReturnForTarget(missingReturn, sink, target, diagnoseWarning);
            }
        }
    }

    for (auto childInst : inst->getDecorationsAndChildren())
    {
        checkForMissingReturnsRec(childInst, sink, target, diagnoseWarning);
    }
}

void checkForMissingReturns(
    IRModule* module,
    DiagnosticSink* sink,
    CodeGenTarget target,
    bool diagnoseWarning)
{
    // Look for any `missingReturn` instructions
    checkForMissingReturnsRec(module->getModuleInst(), sink, target, diagnoseWarning);
}

} // namespace Slang