MIR Optimise - Add attempt at optimising out calls to atomic wrappers

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;