cpplint

masahi · Apr 20, 2022 · e8c4405 · e8c4405
1 parent 7275684
commit e8c4405
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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