diff options
Diffstat (limited to 'source/slang/slang-emit.cpp')
| -rw-r--r-- | source/slang/slang-emit.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/source/slang/slang-emit.cpp b/source/slang/slang-emit.cpp index 58376bbc1..f093104bd 100644 --- a/source/slang/slang-emit.cpp +++ b/source/slang/slang-emit.cpp @@ -90,6 +90,7 @@ #include "slang-ir-specialize-buffer-load-arg.h" #include "slang-ir-specialize-matrix-layout.h" #include "slang-ir-specialize-resources.h" +#include "slang-ir-specialize-stage-switch.h" #include "slang-ir-specialize.h" #include "slang-ir-ssa-simplification.h" #include "slang-ir-ssa.h" @@ -322,6 +323,7 @@ struct RequiredLoweringPassSet bool dynamicResource; bool dynamicResourceHeap; bool resolveVaryingInputRef; + bool specializeStageSwitch; }; // Scan the IR module and determine which lowering/legalization passes are needed based @@ -444,6 +446,9 @@ void calcRequiredLoweringPassSet( case kIROp_ResolveVaryingInputRef: result.resolveVaryingInputRef = true; break; + case kIROp_GetCurrentStage: + result.specializeStageSwitch = true; + break; } if (!result.generics || !result.existentialTypeLayout) { @@ -1027,6 +1032,10 @@ Result linkAndOptimizeIR( cleanupGenerics(targetProgram, irModule, sink); dumpIRIfEnabled(codeGenContext, irModule, "AFTER-LOWER-GENERICS"); + // After dynamic dispatch logic is resolved into ordinary function calls, + // we can now run our stage specialization logic. + if (requiredLoweringPassSet.specializeStageSwitch) + specializeStageSwitch(irModule); if (sink->getErrorCount() != 0) return SLANG_FAIL; #if 0 |