update vnni intrin name

masahi · Apr 7, 2022 · 40f9742 · 40f9742
1 parent 4814f82
commit 40f9742
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/meta_schedule/schedule_rule/multi_level_tiling.cc b/src/meta_schedule/schedule_rule/multi_level_tiling.cc
@@ -449,7 +449,7 @@ class MultiLevelTilingNode : public ScheduleRuleNode {
 inline std::vector<State> MultiLevelTilingNode::TileForVNNI(State state) const {
   std::vector<State> result;
   BlockRV block_rv = state.block_rv;
-  const std::string intrin_name = "dot_16x1x16_uint8_int8_int32_cascadelake";
+  const std::string intrin_name = "dot_16x4_vnni";
   Optional<LoopRV> tiled_loop_rv = TilingwithTensorIntrin(state.sch, block_rv, intrin_name);
   ICHECK(tiled_loop_rv.defined());
   state.block_rv = state.sch->Blockize(tiled_loop_rv.value());