diff --git a/src/meta_schedule/feature_extractor/per_store_feature.cc b/src/meta_schedule/feature_extractor/per_store_feature.cc index 955e39df4fc3..5d91bc34f9ae 100644 --- a/src/meta_schedule/feature_extractor/per_store_feature.cc +++ b/src/meta_schedule/feature_extractor/per_store_feature.cc @@ -279,6 +279,9 @@ Pass SimplifyForFeatureExtraction() { } Stmt VisitStmt_(const ForNode* loop) final { + if (is_zero(loop->extent)) { + return Evaluate(0); + } if (is_zero(loop->min) && is_one(loop->extent) && loop->kind == ForKind::kSerial && loop->annotations.empty()) { unit_vars_.insert(loop->loop_var);