fix passing storage scope

masahi · Jun 29, 2021 · f392e47 · f392e47
1 parent bf132cf
commit f392e47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/te/schedule/schedule_ops.cc b/src/te/schedule/schedule_ops.cc
@@ -214,7 +214,7 @@ class SchedulePostProc : public StmtExprMutator {
     auto it = replace_realize_.find(key);
     if (it != replace_realize_.end()) {
       if (it->second.defined()) {
-        Stmt ret = ProducerRealize(it->second, op->bounds, op->condition, op->body);
+        Stmt ret = ProducerRealize(it->second, op->bounds, op->condition, op->body, op->storage_scope);
         return this->VisitStmt(ret);
       } else {
         return this->VisitStmt(op->body);