@@ -54,20 +54,14 @@ Expr Optimize(Expr e,
5454
5555 auto copied = ir::ir_utils::IRCopy (e);
5656
57- FoldCINNCallArguments (&copied);
58- TransformPolyForToFor (&copied);
5957 ReplaceConstParamToInteger (&copied);
6058 // Simplify already contains CastSimplify
6159 Simplify (&copied);
6260 ReplaceCrossThreadReduction (&copied);
6361 VLOG (4 ) << " After Optimize ReplaceCrossThreadReduction:" << copied;
6462 ReplaceCrossBlockReduction (&copied);
6563 VLOG (4 ) << " After Optimize ReplaceCrossBlockReduction:" << copied;
66- UnrollLoop (&copied);
67- VLOG (4 ) << " After Optimize UnrollLoop:" << copied;
6864
69- VectorizeLoops (&copied, target);
70- VLOG (4 ) << " After Optimize VectorizeLoops:" << copied;
7165 cinn::common::DefaultDeviceTarget ().arch .Match (
7266 [&](std::variant<common::UnknownArch, common::X86Arch, common::ARMArch>) {
7367 },
@@ -120,10 +114,6 @@ Expr Optimize(Expr e,
120114
121115ir::Module Optimize (const ir::Module& module , const Target& target) {
122116 auto copied = ir::ir_utils::IRCopy (Expr (module ));
123- ReplaceCrossThreadReduction (&copied);
124- UnrollLoop (&copied);
125- VectorizeLoops (&copied, Target ());
126- VLOG (10 ) << " After VectorizeLoops:" << copied.as_module_ref ();
127117 RemoveScheduleBlock (&copied);
128118 VLOG (10 ) << " After RemoveScheduleBlock:" << copied.as_module_ref ();
129119 LowerFunctionCallBindVars (&copied);
0 commit comments