add drop and tee expressions

1 files changed, 8 insertions, 0 deletions

diff --git a/src/passes/Vacuum.cpp b/src/passes/Vacuum.cpp
index 0195427d7..1b0887181 100644
--- a/src/passes/Vacuum.cpp
+++ b/src/passes/Vacuum.cpp
@@ -41,6 +41,7 @@ struct Vacuum : public WalkerPass<PostWalker<Vacuum, Visitor<Vacuum>>> {
         case Expression::Id::BlockId: return curr; // not always needed, but handled in visitBlock()
         case Expression::Id::IfId: return curr; // not always needed, but handled in visitIf()
         case Expression::Id::LoopId: return curr; // not always needed, but handled in visitLoop()
+        case Expression::Id::DropId: return curr; // not always needed, but handled in visitDrop()
 
         case Expression::Id::BreakId:
         case Expression::Id::SwitchId:
@@ -198,6 +199,13 @@ struct Vacuum : public WalkerPass<PostWalker<Vacuum, Visitor<Vacuum>>> {
     if (curr->body->is<Nop>()) ExpressionManipulator::nop(curr);
   }
 
+  void visitDrop(Drop* curr) {
+    // if the drop input has no side effects, it can be wiped out
+    if (!EffectAnalyzer(curr->value).hasSideEffects()) {
+      ExpressionManipulator::nop(curr);
+    }
+  }
+
   static void visitPre(Vacuum* self, Expression** currp) {
     self->expressionStack.push_back(*currp);
   }