Erase decl_buffer when it goes out of scope

apache · Aug 30, 2022 · 071b81d · 071b81d
1 parent 6807883
commit 071b81d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/printer/tvmscript_printer.cc b/src/printer/tvmscript_printer.cc
@@ -103,6 +103,7 @@ class BufferUsageFinder : public StmtExprVisitor {
   void VisitStmt_(const DeclBufferNode* op) final {
     buffers_declared_.insert(op->buffer.get());
     StmtExprVisitor::VisitStmt_(op);
+    buffers_declared_.erase(op->buffer.get());
   }
 
  private:
@@ -1069,7 +1070,7 @@ namespace {
 
 bool IsAllocateDeclBufferPattern(const AllocateNode* allocate) {
   const Var& buffer_var = allocate->buffer_var;
-  const DeclBufferNode* decl_buffer = allocat->body.as<DeclBufferNode>();
+  const DeclBufferNode* decl_buffer = allocate->body.as<DeclBufferNode>();
   if (!decl_buffer) {
     return false;
   }