diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/ir/linear-execution.h | 11 | ||||
| -rw-r--r-- | src/passes/SimplifyLocals.cpp | 8 |
2 files changed, 16 insertions, 3 deletions
diff --git a/src/ir/linear-execution.h b/src/ir/linear-execution.h index b1d0b96c3..a37349489 100644 --- a/src/ir/linear-execution.h +++ b/src/ir/linear-execution.h @@ -17,6 +17,7 @@ #ifndef wasm_ir_linear_execution_h #define wasm_ir_linear_execution_h +#include <ir/properties.h> #include <wasm-traversal.h> #include <wasm.h> @@ -126,7 +127,17 @@ struct LinearExecutionWalker : public PostWalker<SubType, VisitorType> { self->pushTask(SubType::doNoteNonLinear, currp); break; } + case Expression::Id::BrOnId: { + self->pushTask(SubType::doVisitBrOn, currp); + self->pushTask(SubType::doNoteNonLinear, currp); + self->maybePushTask(SubType::scan, &curr->cast<BrOn>()->rtt); + self->pushTask(SubType::scan, &curr->cast<BrOn>()->ref); + break; + } default: { + // All relevant things should have been handled. + assert(!Properties::isControlFlowStructure(curr)); + assert(!Properties::isBranch(curr)); // other node types do not have control flow, use regular post-order PostWalker<SubType, VisitorType>::scan(self, currp); } diff --git a/src/passes/SimplifyLocals.cpp b/src/passes/SimplifyLocals.cpp index 02be1ceec..77b8f1a93 100644 --- a/src/passes/SimplifyLocals.cpp +++ b/src/passes/SimplifyLocals.cpp @@ -140,9 +140,11 @@ struct SimplifyLocals } else if (curr->is<If>()) { assert(!curr->cast<If>() ->ifFalse); // if-elses are handled by doNoteIf* methods - } else if (curr->is<Switch>()) { - auto* sw = curr->cast<Switch>(); - auto targets = BranchUtils::getUniqueTargets(sw); + } else { + // Not one of the recognized instructions, so do not optimize here: mark + // all the targets as unoptimizable. + // TODO optimize BrOn, Switch, etc. + auto targets = BranchUtils::getUniqueTargets(curr); for (auto target : targets) { self->unoptimizableBlocks.insert(target); } |