summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/mir/optimise.cpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/src/mir/optimise.cpp b/src/mir/optimise.cpp
index a07e1a38..9935454e 100644
--- a/src/mir/optimise.cpp
+++ b/src/mir/optimise.cpp
@@ -794,6 +794,15 @@ bool MIR_Optimise_Inlining(::MIR::TypeResolve& state, ::MIR::Function& fcn, bool
return false;
return true;
}
+ else if( fcn.blocks.size() > 1 && fcn.blocks[0].terminator.is_Switch() )
+ {
+ // Setup + Arms + Return + Panic
+ // - Handles the atomit wrappers
+ if( fcn.blocks.size() != fcn.blocks[0].terminator.as_Switch().targets.size()+3 )
+ return false;
+ // TODO: Check for the parameter being a Constant?
+ return true;
+ }
else
{
return false;