diff --git a/src/meta_schedule/schedule_rule/auto_tensorize.cc b/src/meta_schedule/schedule_rule/auto_tensorize.cc
index fb709bf52131a..d2c1deaf632a7 100644
--- a/src/meta_schedule/schedule_rule/auto_tensorize.cc
+++ b/src/meta_schedule/schedule_rule/auto_tensorize.cc
@@ -19,6 +19,9 @@
 
 #include "auto_tensorize.h"
 
+#include <unordered_set>
+#include <vector>
+
 #include "../../tir/schedule/analysis.h"
 
 namespace tvm {
@@ -97,6 +100,5 @@ tir::BlockRV TileForIntrin(tir::Schedule sch, tir::BlockRV block, const std::str
 
 TVM_REGISTER_GLOBAL("meta_schedule.TilingwithTensorIntrin").set_body_typed(TilingwithTensorIntrin);
 
-
 }  // namespace meta_schedule
 }  // namespace tvm