diff options
Diffstat (limited to 'test/libjulia/YulOptimizerTest.cpp')
| -rw-r--r-- | test/libjulia/YulOptimizerTest.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/test/libjulia/YulOptimizerTest.cpp b/test/libjulia/YulOptimizerTest.cpp index 9f028bcd..0e55c87d 100644 --- a/test/libjulia/YulOptimizerTest.cpp +++ b/test/libjulia/YulOptimizerTest.cpp @@ -29,6 +29,7 @@ #include <libjulia/optimiser/FunctionHoister.h> #include <libjulia/optimiser/ExpressionInliner.h> #include <libjulia/optimiser/FullInliner.h> +#include <libjulia/optimiser/MainFunction.h> #include <libsolidity/parsing/Scanner.h> #include <libsolidity/inlineasm/AsmPrinter.h> @@ -121,6 +122,12 @@ bool YulOptimizerTest::run(ostream& _stream, string const& _linePrefix, bool con (FunctionGrouper{})(*m_ast); FullInliner(*m_ast).run(); } + else if (m_optimizerStep == "mainFunction") + { + disambiguate(); + (FunctionGrouper{})(*m_ast); + (MainFunction{})(*m_ast); + } else { FormattedScope(_stream, _formatted, {formatting::BOLD, formatting::RED}) << _linePrefix << "Invalid optimizer step: " << m_optimizerStep << endl; |