diff --git a/src/transform/auto_schedule/memory_detector.h b/src/transform/auto_schedule/memory_detector.h index 8a37fc6eea..2f9127f82a 100644 --- a/src/transform/auto_schedule/memory_detector.h +++ b/src/transform/auto_schedule/memory_detector.h @@ -264,6 +264,7 @@ class MemoryAccessDetector : public StmtExprVisitor { } void VisitStmt_(const LetStmtNode *op) override { + VisitExpr(op->value); let_bindings_[op->var.get()] = op->value; UpdateWriteVar(op->var); StmtExprVisitor::VisitStmt(op->body);