diff options
Diffstat (limited to 'src/wasm-traversal.h')
| -rw-r--r-- | src/wasm-traversal.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/wasm-traversal.h b/src/wasm-traversal.h index 3b1de2e32..11332b5c3 100644 --- a/src/wasm-traversal.h +++ b/src/wasm-traversal.h @@ -49,6 +49,7 @@ struct Visitor { ReturnType visitSetGlobal(SetGlobal* curr) {} ReturnType visitLoad(Load* curr) {} ReturnType visitStore(Store* curr) {} + ReturnType visitAtomicRMW(AtomicRMW* curr) {return ReturnType();} //Stub impl so not every pass has to implement this yet. ReturnType visitConst(Const* curr) {} ReturnType visitUnary(Unary* curr) {} ReturnType visitBinary(Binary* curr) {} @@ -90,6 +91,7 @@ struct Visitor { case Expression::Id::SetGlobalId: DELEGATE(SetGlobal); case Expression::Id::LoadId: DELEGATE(Load); case Expression::Id::StoreId: DELEGATE(Store); + case Expression::Id::AtomicRMWId: DELEGATE(AtomicRMW); case Expression::Id::ConstId: DELEGATE(Const); case Expression::Id::UnaryId: DELEGATE(Unary); case Expression::Id::BinaryId: DELEGATE(Binary); @@ -130,6 +132,7 @@ struct UnifiedExpressionVisitor : public Visitor<SubType> { ReturnType visitSetGlobal(SetGlobal* curr) { return static_cast<SubType*>(this)->visitExpression(curr); } ReturnType visitLoad(Load* curr) { return static_cast<SubType*>(this)->visitExpression(curr); } ReturnType visitStore(Store* curr) { return static_cast<SubType*>(this)->visitExpression(curr); } + ReturnType visitAtomicRMW(AtomicRMW* curr) { return static_cast<SubType*>(this)->visitExpression(curr); } ReturnType visitConst(Const* curr) { return static_cast<SubType*>(this)->visitExpression(curr); } ReturnType visitUnary(Unary* curr) { return static_cast<SubType*>(this)->visitExpression(curr); } ReturnType visitBinary(Binary* curr) { return static_cast<SubType*>(this)->visitExpression(curr); } @@ -306,6 +309,7 @@ struct Walker : public VisitorType { static void doVisitSetGlobal(SubType* self, Expression** currp) { self->visitSetGlobal((*currp)->cast<SetGlobal>()); } static void doVisitLoad(SubType* self, Expression** currp) { self->visitLoad((*currp)->cast<Load>()); } static void doVisitStore(SubType* self, Expression** currp) { self->visitStore((*currp)->cast<Store>()); } + static void doVisitAtomicRMW(SubType* self, Expression** currp) { self->visitAtomicRMW((*currp)->cast<AtomicRMW>()); } static void doVisitConst(SubType* self, Expression** currp) { self->visitConst((*currp)->cast<Const>()); } static void doVisitUnary(SubType* self, Expression** currp) { self->visitUnary((*currp)->cast<Unary>()); } static void doVisitBinary(SubType* self, Expression** currp) { self->visitBinary((*currp)->cast<Binary>()); } @@ -428,6 +432,12 @@ struct PostWalker : public Walker<SubType, VisitorType> { self->pushTask(SubType::scan, &curr->cast<Store>()->ptr); break; } + case Expression::Id::AtomicRMWId: { + self->pushTask(SubType::doVisitAtomicRMW, currp); + self->pushTask(SubType::scan, &curr->cast<AtomicRMW>()->value); + self->pushTask(SubType::scan, &curr->cast<AtomicRMW>()->ptr); + break; + } case Expression::Id::ConstId: { self->pushTask(SubType::doVisitConst, currp); break; |